Maria commited on
Commit
5cbab77
·
1 Parent(s): 79dfcb1
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ def infer(
43
  unet_sub_dir = os.path.join(LoRA_path, "unet")
44
  text_encoder_sub_dir = os.path.join(LoRA_path, "text_encoder")
45
 
46
- pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype).to(device)
47
  pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir, adapter_name=adapter_name)
48
 
49
  pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, text_encoder_sub_dir, adapter_name=adapter_name)
 
43
  unet_sub_dir = os.path.join(LoRA_path, "unet")
44
  text_encoder_sub_dir = os.path.join(LoRA_path, "text_encoder")
45
 
46
+ pipe = DiffusionPipeline.from_pretrained('CompVis/stable-diffusion-v1-4', torch_dtype=torch_dtype).to(device)
47
  pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir, adapter_name=adapter_name)
48
 
49
  pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, text_encoder_sub_dir, adapter_name=adapter_name)