hw6

infer.py

@@ -34,7 +34,7 @@ else:
     torch_dtype = torch.float32
 
 default_model = 'CompVis/stable-diffusion-v1-4'
-LoRA_path = 'old_model'
+LoRA_path = 'new_model'
 
 CONTROLNET_MODE = {
     "Canny Edge Detection" : "lllyasviel/control_v11p_sd15_canny",
@@ -118,7 +118,7 @@ def get_pipe(
 
 
     if model_id == 'Maria_Lashina_LoRA':
-        adapter_name = 'a cartoonish mouse'
+        adapter_name = 'cartoonish mouse'
         unet_sub_dir = os.path.join(LoRA_path, "unet")
         text_encoder_sub_dir = os.path.join(LoRA_path, "text_encoder")