Abhi5ingh commited on
Commit
71862df
1 Parent(s): bdbab61

Update sdfile.py

Browse files

Changed the ControlNet model to the latest checkpoint

Files changed (1) hide show
  1. sdfile.py +1 -1
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/model_dresscode", 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))
 
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))