RiddleHe
/

SD14_pathology_controlnet

stable-diffusion

stable-diffusion-diffusers

diffusers-training

Model card Files Files and versions

RiddleHe commited on Feb 6

Commit

d0267aa

·

verified ·

1 Parent(s): 6eb9d1b

Update README.md

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -37,9 +37,8 @@ prompt: An image of breast cancer histopathology with detailed cellular structur
 ```python
 controlnet = ControlNetModel.from_pretrained("RiddleHe/SD14_pathology_controlnet", torch_dtype=torch.float16)
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
-    "CompVis/stable-diffusion-v1-4", controlnet=controlnet, torch_dtype=torch.float16
 )
-pipe.load_lora_weights("RiddleHe/SD14_pathology_lora")  # Load lora weights for pathology image generation
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.to('cuda')

 ```python
 controlnet = ControlNetModel.from_pretrained("RiddleHe/SD14_pathology_controlnet", torch_dtype=torch.float16)
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
+    "RiddleHe/SD14_pathology_base", controlnet=controlnet, torch_dtype=torch.float16
 )
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.to('cuda')