Spaces:

weepiess2383
/

CFG-Zero-Star

Running on Zero

WeichenFan commited on Mar 27

Commit

0cbc98b

1 Parent(s): 66d92f2

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -50,7 +50,7 @@ def load_model(model_name):
     return current_model.to("cuda")
-# @spaces.GPU(duration=200)
 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):

     return current_model.to("cuda")
+@spaces.GPU(duration=120)
 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):