Update sdfile.py
Browse files
sdfile.py
CHANGED
|
@@ -25,7 +25,7 @@ def get_pipelines( name:PIPELINES, enable_cpu_offload = False, ) -> StableDiffus
|
|
| 25 |
pipe.unet.load_attn_procs("./")
|
| 26 |
pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
|
| 27 |
elif name == "sketch2img":
|
| 28 |
-
controlnet = ControlNetModel.from_pretrained("Abhi5ingh/
|
| 29 |
pipe = StableDiffusionControlNetPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", controlnet = controlnet, torch_dtype = torch.float16,cache_dir="D:/huggingface/CACHE/")
|
| 30 |
pipe.unet.load_attn_procs("./")
|
| 31 |
pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
|
|
|
|
| 25 |
pipe.unet.load_attn_procs("./")
|
| 26 |
pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
|
| 27 |
elif name == "sketch2img":
|
| 28 |
+
controlnet = ControlNetModel.from_pretrained("Abhi5ingh/ControlnetDresscode", torch_dtype=torch.float16,cache_dir="D:/huggingface/CACHE/")
|
| 29 |
pipe = StableDiffusionControlNetPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", controlnet = controlnet, torch_dtype = torch.float16,cache_dir="D:/huggingface/CACHE/")
|
| 30 |
pipe.unet.load_attn_procs("./")
|
| 31 |
pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
|