ironjr commited on
Commit
3dd5d26
1 Parent(s): 919599a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -82,7 +82,8 @@ parser.add_argument('-H', '--height', type=int, default=768)
82
  parser.add_argument('-W', '--width', type=int, default=768)
83
  parser.add_argument('--model', type=str, default=None, help='Hugging face model repository or local path for a SD1.5 model checkpoint to run.')
84
  parser.add_argument('--bootstrap_steps', type=int, default=1)
85
- parser.add_argument('--guidance_scale', type=float, default=0) # 1.2
 
86
  parser.add_argument('--run_time', type=float, default=60)
87
  parser.add_argument('--seed', type=int, default=-1)
88
  parser.add_argument('--device', type=int, default=0)
@@ -114,7 +115,7 @@ model = StreamMultiDiffusion(
114
  use_tiny_vae=True,
115
  mask_type='continuous',
116
  bootstrap_steps=opt.bootstrap_steps,
117
- bootstrap_mix_steps=opt.bootstrap_steps,
118
  guidance_scale=opt.guidance_scale,
119
  seed=opt.seed,
120
  ).cuda()
 
82
  parser.add_argument('-W', '--width', type=int, default=768)
83
  parser.add_argument('--model', type=str, default=None, help='Hugging face model repository or local path for a SD1.5 model checkpoint to run.')
84
  parser.add_argument('--bootstrap_steps', type=int, default=1)
85
+ parser.add_argument('--bootstrap_mix_steps', type=int, default=0)
86
+ parser.add_argument('--guidance_scale', type=float, default=1.0) # 1.2
87
  parser.add_argument('--run_time', type=float, default=60)
88
  parser.add_argument('--seed', type=int, default=-1)
89
  parser.add_argument('--device', type=int, default=0)
 
115
  use_tiny_vae=True,
116
  mask_type='continuous',
117
  bootstrap_steps=opt.bootstrap_steps,
118
+ bootstrap_mix_steps=opt.bootstrap_mix_steps,
119
  guidance_scale=opt.guidance_scale,
120
  seed=opt.seed,
121
  ).cuda()