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)
|