Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -40,7 +40,7 @@ def load_model( progress=None):
|
|
| 40 |
if progress is not None:
|
| 41 |
progress(0.1, desc=f"Loading model from {MODEL_REPO_ID}...")
|
| 42 |
|
| 43 |
-
time_scale =
|
| 44 |
|
| 45 |
# Load pipeline (you had bfloat16 here; keep if you like)
|
| 46 |
pipe = SiDSanaPipeline.from_pretrained(MODEL_REPO_ID, torch_dtype=torch_dtype)
|
|
@@ -144,14 +144,14 @@ with gr.Blocks(css=css) as demo:
|
|
| 144 |
minimum=256,
|
| 145 |
maximum=MAX_IMAGE_SIZE,
|
| 146 |
step=32,
|
| 147 |
-
value=
|
| 148 |
)
|
| 149 |
height = gr.Slider(
|
| 150 |
label="Height",
|
| 151 |
minimum=256,
|
| 152 |
maximum=MAX_IMAGE_SIZE,
|
| 153 |
step=32,
|
| 154 |
-
value=
|
| 155 |
)
|
| 156 |
|
| 157 |
with gr.Row():
|
|
|
|
| 40 |
if progress is not None:
|
| 41 |
progress(0.1, desc=f"Loading model from {MODEL_REPO_ID}...")
|
| 42 |
|
| 43 |
+
time_scale = 1.0 # for SANA Rectified Flow / TrigFlow
|
| 44 |
|
| 45 |
# Load pipeline (you had bfloat16 here; keep if you like)
|
| 46 |
pipe = SiDSanaPipeline.from_pretrained(MODEL_REPO_ID, torch_dtype=torch_dtype)
|
|
|
|
| 144 |
minimum=256,
|
| 145 |
maximum=MAX_IMAGE_SIZE,
|
| 146 |
step=32,
|
| 147 |
+
value=1024,
|
| 148 |
)
|
| 149 |
height = gr.Slider(
|
| 150 |
label="Height",
|
| 151 |
minimum=256,
|
| 152 |
maximum=MAX_IMAGE_SIZE,
|
| 153 |
step=32,
|
| 154 |
+
value=1024,
|
| 155 |
)
|
| 156 |
|
| 157 |
with gr.Row():
|