Spaces:

mariarivaille
/

Diffusion_Models_Course_Space

Sleeping

Maria commited on Mar 8

Commit

df64a69

1 Parent(s): bcad7c8

hw6

Files changed (1) hide show

infer.py CHANGED Viewed

@@ -58,8 +58,7 @@ def get_pipe(
         controlnet = ControlNetModel.from_pretrained(
             CONTROLNET_MODE[controlnet_mode],
-            cache_dir="./models_cache",
-            torch_dtype=torch.float16
         )
         pipe = StableDiffusionControlNetPipeline.from_pretrained(
@@ -81,8 +80,8 @@ def get_pipe(
         controlnet = ControlNetModel.from_pretrained(
             CONTROLNET_MODE[controlnet_mode],
-            cache_dir="./models_cache",
-            torch_dtype=torch.float16)
         pipe = StableDiffusionControlNetPipeline.from_pretrained(
             model_id if model_id!='Maria_Lashina_LoRA' else default_model,

         controlnet = ControlNetModel.from_pretrained(
             CONTROLNET_MODE[controlnet_mode],
+            cache_dir="./models_cache"
         )
         pipe = StableDiffusionControlNetPipeline.from_pretrained(
         controlnet = ControlNetModel.from_pretrained(
             CONTROLNET_MODE[controlnet_mode],
+            cache_dir="./models_cache"
+            )
         pipe = StableDiffusionControlNetPipeline.from_pretrained(
             model_id if model_id!='Maria_Lashina_LoRA' else default_model,