Maria
commited on
Commit
·
5cbab77
1
Parent(s):
79dfcb1
fixed
Browse files
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(
|
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)
|