Spaces:
Runtime error
Runtime error
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()
|