Commit
•
0b93385
1
Parent(s):
63717de
Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ def update_selection(evt: gr.SelectData):
|
|
47 |
evt.index
|
48 |
)
|
49 |
|
50 |
-
@spaces.GPU(duration=
|
51 |
def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, progress):
|
52 |
pipe.to("cuda")
|
53 |
generator = torch.Generator(device="cuda").manual_seed(seed)
|
@@ -90,6 +90,8 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
|
|
90 |
pipe.unload_lora_weights()
|
91 |
return image, seed
|
92 |
|
|
|
|
|
93 |
css = '''
|
94 |
#gen_btn{height: 100%}
|
95 |
#title{text-align: center;}
|
|
|
47 |
evt.index
|
48 |
)
|
49 |
|
50 |
+
@spaces.GPU(duration=70)
|
51 |
def generate_image(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, progress):
|
52 |
pipe.to("cuda")
|
53 |
generator = torch.Generator(device="cuda").manual_seed(seed)
|
|
|
90 |
pipe.unload_lora_weights()
|
91 |
return image, seed
|
92 |
|
93 |
+
run_lora.zerogpu = True
|
94 |
+
|
95 |
css = '''
|
96 |
#gen_btn{height: 100%}
|
97 |
#title{text-align: center;}
|