Update app.py
Browse files
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=
|
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 =
|
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
|