PommesPeter commited on
Commit
17fcb1c
·
verified ·
1 Parent(s): 0f6358a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -2
app.py CHANGED
@@ -194,7 +194,7 @@ def infer_ode(args, infer_args, text_encoder, tokenizer, vae, model):
194
  solver,
195
  t_shift,
196
  seed,
197
- scale_method,
198
  proportional_attn,
199
  ) = infer_args
200
 
@@ -207,7 +207,7 @@ def infer_ode(args, infer_args, text_encoder, tokenizer, vae, model):
207
  solver=solver,
208
  t_shift=t_shift,
209
  seed=seed,
210
- ntk_scaling=scale_method,
211
  proportional_attn=proportional_attn,
212
  )
213
  print("> params:", json.dumps(metadata, indent=2))
@@ -521,6 +521,23 @@ def main():
521
  )
522
  with gr.Row():
523
  submit_btn = gr.Button("Submit", variant="primary")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
524
  with gr.Column():
525
  output_img = gr.Image(
526
  label="Lumina Generated image",
 
194
  solver,
195
  t_shift,
196
  seed,
197
+ scaling_method,
198
  proportional_attn,
199
  ) = infer_args
200
 
 
207
  solver=solver,
208
  t_shift=t_shift,
209
  seed=seed,
210
+ scaling_method=scaling_method,
211
  proportional_attn=proportional_attn,
212
  )
213
  print("> params:", json.dumps(metadata, indent=2))
 
521
  )
522
  with gr.Row():
523
  submit_btn = gr.Button("Submit", variant="primary")
524
+ reset_btn = gr.ClearButton(
525
+ [
526
+ cap,
527
+ neg_cap,
528
+ resolution,
529
+ num_sampling_steps,
530
+ cfg_scale,
531
+ solver,
532
+ t_shift,
533
+ seed,
534
+ scaling_method,
535
+ proportional_attn,
536
+ ],
537
+ value="Cancel",
538
+ variant="stop",
539
+
540
+ )
541
  with gr.Column():
542
  output_img = gr.Image(
543
  label="Lumina Generated image",