Spaces:
Running
Running
Commit
•
aac3460
1
Parent(s):
4f76251
Update app.py
Browse files
app.py
CHANGED
@@ -186,7 +186,7 @@ from diffusers import AutoPipelineForText2Image
|
|
186 |
import torch
|
187 |
|
188 |
pipeline = AutoPipelineForText2Image.from_pretrained('{info["baseModel"]}', torch_dtype=torch.float16).to('cuda')
|
189 |
-
pipeline.load_lora_weights(
|
190 |
image = pipeline('{prompt if prompt else (formatted_words if formatted_words else 'Your custom prompt')}').images[0]
|
191 |
```
|
192 |
|
|
|
186 |
import torch
|
187 |
|
188 |
pipeline = AutoPipelineForText2Image.from_pretrained('{info["baseModel"]}', torch_dtype=torch.float16).to('cuda')
|
189 |
+
pipeline.load_lora_weights('{user_repo_id}', weight_name='{downloaded_files["weightName"]}')
|
190 |
image = pipeline('{prompt if prompt else (formatted_words if formatted_words else 'Your custom prompt')}').images[0]
|
191 |
```
|
192 |
|