multimodalart HF staff commited on
Commit
aac3460
1 Parent(s): 4f76251

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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("{{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
 
 
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