Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
84e6fa0
1
Parent(s):
a8a382e
Update app.py
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ css = '''
|
|
43 |
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
|
44 |
original_pipe = copy.deepcopy(pipe)
|
45 |
|
46 |
-
|
47 |
def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lora_2_scale=0.5, progress=gr.Progress(track_tqdm=True)):
|
48 |
pipe = copy.deepcopy(original_pipe)
|
49 |
pipe.load_lora_weights(shuffled_items[0]['repo'], weight_name=shuffled_items[0]['weights'])
|
|
|
43 |
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
|
44 |
original_pipe = copy.deepcopy(pipe)
|
45 |
|
46 |
+
#@spaces.GPU
|
47 |
def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lora_2_scale=0.5, progress=gr.Progress(track_tqdm=True)):
|
48 |
pipe = copy.deepcopy(original_pipe)
|
49 |
pipe.load_lora_weights(shuffled_items[0]['repo'], weight_name=shuffled_items[0]['weights'])
|