Spaces:
Running
on
Zero
Running
on
Zero
setting parameters instead of hardcoding it
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ MAX_SEED = np.iinfo(np.int16).max
|
|
22 |
MAX_IMAGE_SIZE = 512
|
23 |
|
24 |
DEFAULT_POSITIVE_PROMPT = None
|
25 |
-
DEFAULT_NEGATIVE_PROMPT =
|
26 |
|
27 |
@spaces.GPU(duration=300)
|
28 |
def infer(
|
@@ -31,7 +31,7 @@ def infer(
|
|
31 |
seed=0,
|
32 |
text_cfg=7.5,
|
33 |
img_cfg=1.0,
|
34 |
-
num_inference_steps=
|
35 |
positive_prompt=DEFAULT_POSITIVE_PROMPT,
|
36 |
negative_prompt=DEFAULT_NEGATIVE_PROMPT,
|
37 |
progress=gr.Progress(track_tqdm=True),
|
@@ -56,7 +56,7 @@ def infer(
|
|
56 |
use_norm=False,
|
57 |
num_sampling_steps=num_inference_steps,
|
58 |
timesteps_shift=1.0,
|
59 |
-
seed=
|
60 |
progress=True,
|
61 |
)
|
62 |
|
@@ -117,7 +117,7 @@ with gr.Blocks(css=css) as demo:
|
|
117 |
minimum=10,
|
118 |
maximum=50,
|
119 |
step=1,
|
120 |
-
value=
|
121 |
)
|
122 |
|
123 |
with gr.Row():
|
|
|
22 |
MAX_IMAGE_SIZE = 512
|
23 |
|
24 |
DEFAULT_POSITIVE_PROMPT = None
|
25 |
+
DEFAULT_NEGATIVE_PROMPT = None
|
26 |
|
27 |
@spaces.GPU(duration=300)
|
28 |
def infer(
|
|
|
31 |
seed=0,
|
32 |
text_cfg=7.5,
|
33 |
img_cfg=1.0,
|
34 |
+
num_inference_steps=28,
|
35 |
positive_prompt=DEFAULT_POSITIVE_PROMPT,
|
36 |
negative_prompt=DEFAULT_NEGATIVE_PROMPT,
|
37 |
progress=gr.Progress(track_tqdm=True),
|
|
|
56 |
use_norm=False,
|
57 |
num_sampling_steps=num_inference_steps,
|
58 |
timesteps_shift=1.0,
|
59 |
+
seed=seed,
|
60 |
progress=True,
|
61 |
)
|
62 |
|
|
|
117 |
minimum=10,
|
118 |
maximum=50,
|
119 |
step=1,
|
120 |
+
value=28, # Replace with defaults that work for your model
|
121 |
)
|
122 |
|
123 |
with gr.Row():
|