multimodalart HF staff commited on
Commit
4d2f098
·
1 Parent(s): efbdbfe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -93,7 +93,7 @@ div#share-btn-container > div {flex-direction: row;background: black;align-items
93
  original_pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
94
 
95
  @spaces.GPU
96
- def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lora_2_scale=0.5, seed=-1):
97
 
98
  repo_id_1 = shuffled_items[0]['repo']
99
  repo_id_2 = shuffled_items[1]['repo']
 
93
  original_pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
94
 
95
  @spaces.GPU
96
+ def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lora_2_scale=0.5, seed=-1, progress=gr.Progress(track_tqdm=True)):
97
 
98
  repo_id_1 = shuffled_items[0]['repo']
99
  repo_id_2 = shuffled_items[1]['repo']