Update sdfile.py
Browse filesChanged the ControlNet model to the latest checkpoint
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))
|