Spaces:
Running
on
Zero
Running
on
Zero
WeichenFan
commited on
Commit
·
66d92f2
1
Parent(s):
453d861
update
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ def load_model(model_name):
|
|
| 50 |
|
| 51 |
return current_model.to("cuda")
|
| 52 |
|
| 53 |
-
@spaces.GPU(duration=200)
|
| 54 |
def generate_content(prompt, model_name, guidance_scale=7.5, num_inference_steps=50,
|
| 55 |
use_cfg_zero_star=True, use_zero_init=True, zero_steps=0,
|
| 56 |
seed=None, compare_mode=False):
|
|
|
|
| 50 |
|
| 51 |
return current_model.to("cuda")
|
| 52 |
|
| 53 |
+
# @spaces.GPU(duration=200)
|
| 54 |
def generate_content(prompt, model_name, guidance_scale=7.5, num_inference_steps=50,
|
| 55 |
use_cfg_zero_star=True, use_zero_init=True, zero_steps=0,
|
| 56 |
seed=None, compare_mode=False):
|