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'])
|