Spaces:

mariarivaille
/

Diffusion_Models_Course_Space

Sleeping

Maria commited on Mar 10

Commit

56a3b69

1 Parent(s): 129880c

hw6

Files changed (1) hide show

infer.py CHANGED Viewed

@@ -118,7 +118,7 @@ def get_pipe(
     if model_id == 'Maria_Lashina_LoRA':
-        adapter_name = 'a cartoonish mouse'
         unet_sub_dir = os.path.join(LoRA_path, "unet")
         text_encoder_sub_dir = os.path.join(LoRA_path, "text_encoder")

     if model_id == 'Maria_Lashina_LoRA':
+        adapter_name = 'cartoonish mouse'
         unet_sub_dir = os.path.join(LoRA_path, "unet")
         text_encoder_sub_dir = os.path.join(LoRA_path, "text_encoder")