Update sdfile.py

sdfile.py

@@ -21,12 +21,12 @@ def get_pipelines( name:PIPELINES, enable_cpu_offload = False, ) -> StableDiffus
     pipe = None
 
     if name == "txt2img":
-        pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
+        pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16,cache_dir="D:/huggingface/CACHE/")
         pipe.unet.load_attn_procs("./")
         pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
     elif name == "sketch2img":
-        controlnet = ControlNetModel.from_pretrained("Abhi5ingh/model_dresscode", torch_dtype=torch.float16)
-        pipe = StableDiffusionControlNetPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", controlnet = controlnet, torch_dtype = torch.float16)
+        controlnet = ControlNetModel.from_pretrained("Abhi5ingh/model_dresscode", torch_dtype=torch.float16,cache_dir="D:/huggingface/CACHE/")
+        pipe = StableDiffusionControlNetPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", controlnet = controlnet, torch_dtype = torch.float16,cache_dir="D:/huggingface/CACHE/")
         pipe.unet.load_attn_procs("./")
         pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
 
@@ -73,6 +73,7 @@ def generate(
     else:
         raise Exception(
             f"Cannot generate image for pipeline {pipeline_name} and {prompt}")
+    images = pipe(**kwargs).images
     image = images[0]
 
     os.makedirs("outputs", exist_ok=True)
@@ -84,7 +85,5 @@ def generate(
     )
     image.save(f"{filename}.png")
     with open(f"{filename}.txt", "w") as f:
-        f.write(f"Prompt: {prompt}\n\nNegative Prompt:{negative_prompt}"
+        f.write(f"Prompt: {prompt}\n\nNegative Prompt:{negative_prompt}")
     return image
-
-)