YGu1998 commited on
Commit
a2aae93
·
verified ·
1 Parent(s): 80917c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 = 1000.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,14 +144,14 @@ with gr.Blocks(css=css) as demo:
144
  minimum=256,
145
  maximum=MAX_IMAGE_SIZE,
146
  step=32,
147
- value=512,
148
  )
149
  height = gr.Slider(
150
  label="Height",
151
  minimum=256,
152
  maximum=MAX_IMAGE_SIZE,
153
  step=32,
154
- value=512,
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():