Hugging Face's logo

RiddleHe
/
SD14_pathology_controlnet

Text-to-Image
Diffusers
Safetensors
stable-diffusion
stable-diffusion-diffusers
controlnet
diffusers-training
Model card Files
xet
 Community
RiddleHe commited on
Commit
d0267aa
·
verified ·
1 Parent(s): 6eb9d1b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -37,9 +37,8 @@ prompt: An image of breast cancer histopathology with detailed cellular structur
37
  ```python
38
  controlnet = ControlNetModel.from_pretrained("RiddleHe/SD14_pathology_controlnet", torch_dtype=torch.float16)
39
  pipe = StableDiffusionControlNetPipeline.from_pretrained(
40
- "CompVis/stable-diffusion-v1-4", controlnet=controlnet, torch_dtype=torch.float16
41
  )
42
- pipe.load_lora_weights("RiddleHe/SD14_pathology_lora") # Load lora weights for pathology image generation
43
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
44
  pipe.to('cuda')
45
 
 
37
  ```python
38
  controlnet = ControlNetModel.from_pretrained("RiddleHe/SD14_pathology_controlnet", torch_dtype=torch.float16)
39
  pipe = StableDiffusionControlNetPipeline.from_pretrained(
40
+ "RiddleHe/SD14_pathology_base", controlnet=controlnet, torch_dtype=torch.float16
41
  )
 
42
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
43
  pipe.to('cuda')
44