Spaces:
Running
on
A10G
Running
on
A10G
Update app.py
Browse files
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('--
|
|
|
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.
|
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()
|