Spaces:
Sleeping
Sleeping
Commit
•
d9a8a67
1
Parent(s):
af5ea8a
Update app.py
Browse files
app.py
CHANGED
@@ -81,7 +81,7 @@ div#share-btn-container > div {flex-direction: row;background: black;align-items
|
|
81 |
|
82 |
original_pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
|
83 |
|
84 |
-
def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lora_2_scale=0.5, seed, progress=gr.Progress(track_tqdm=True)):
|
85 |
state_dict_1 = copy.deepcopy(shuffled_items[0]['state_dict'])
|
86 |
state_dict_2 = copy.deepcopy(shuffled_items[1]['state_dict'])
|
87 |
pipe = copy.deepcopy(original_pipe)
|
|
|
81 |
|
82 |
original_pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
|
83 |
|
84 |
+
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)):
|
85 |
state_dict_1 = copy.deepcopy(shuffled_items[0]['state_dict'])
|
86 |
state_dict_2 = copy.deepcopy(shuffled_items[1]['state_dict'])
|
87 |
pipe = copy.deepcopy(original_pipe)
|