Spaces:
Running
on
Zero
Running
on
Zero
WeichenFan
commited on
Commit
·
fd45f29
1
Parent(s):
6c06630
Add application file
Browse files
app.py
CHANGED
|
@@ -58,7 +58,7 @@ def load_model(model_name):
|
|
| 58 |
return current_model.to('cuda')
|
| 59 |
|
| 60 |
|
| 61 |
-
@spaces.GPU(duration=
|
| 62 |
def generate_content(prompt, model_name, guidance_scale=7.5, num_inference_steps=50, use_cfg_zero_star=True, use_zero_init=True, zero_steps=0, seed=None, compare_mode=False):
|
| 63 |
model = load_model(model_name)
|
| 64 |
if seed is None:
|
|
|
|
| 58 |
return current_model.to('cuda')
|
| 59 |
|
| 60 |
|
| 61 |
+
@spaces.GPU(duration=1000)
|
| 62 |
def generate_content(prompt, model_name, guidance_scale=7.5, num_inference_steps=50, use_cfg_zero_star=True, use_zero_init=True, zero_steps=0, seed=None, compare_mode=False):
|
| 63 |
model = load_model(model_name)
|
| 64 |
if seed is None:
|