Staticaliza commited on
Commit
13beffb
·
verified ·
1 Parent(s): 1d065cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -69,7 +69,7 @@ def get_seed(seed):
69
  else:
70
  return random.randint(0, MAX_SEED)
71
 
72
- @spaces.GPU(duration=60)
73
  def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATIVE_INPUT, model=DEFAULT_MODEL, height=DEFAULT_HEIGHT, width=DEFAULT_WIDTH, steps=1, guidance=0, number=1, seed=None):
74
 
75
  repo = repo_customs[model or "Default"]
@@ -92,7 +92,7 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
92
  guidance_set = 1.5
93
  repo.set_adapters(["pixel_base", "pixel_base_2"], adapter_weights=[1, 1])
94
  elif model == "Pro":
95
- steps_set = 25
96
  guidance_set = 3.5
97
  else:
98
  steps_set = 25
 
69
  else:
70
  return random.randint(0, MAX_SEED)
71
 
72
+ @spaces.GPU(duration=120)
73
  def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATIVE_INPUT, model=DEFAULT_MODEL, height=DEFAULT_HEIGHT, width=DEFAULT_WIDTH, steps=1, guidance=0, number=1, seed=None):
74
 
75
  repo = repo_customs[model or "Default"]
 
92
  guidance_set = 1.5
93
  repo.set_adapters(["pixel_base", "pixel_base_2"], adapter_weights=[1, 1])
94
  elif model == "Pro":
95
+ steps_set = 15
96
  guidance_set = 3.5
97
  else:
98
  steps_set = 25