Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
4d2f098
1
Parent(s):
efbdbfe
Update app.py
Browse files
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']
|