Spaces:
Running
on
L4
Running
on
L4
FrozenBurning
commited on
Commit
•
02e04ed
1
Parent(s):
2d394ea
fix app.py
Browse files
app.py
CHANGED
|
@@ -84,6 +84,7 @@ def process(input_image, input_num_steps=25, input_seed=42, input_cfg=6.0):
|
|
| 84 |
output_mat_video_path = os.path.join(config.output_dir, GRADIO_MAT_VIDEO_PATH)
|
| 85 |
output_glb_path = os.path.join(config.output_dir, GRADIO_GLB_PATH)
|
| 86 |
|
|
|
|
| 87 |
diffusion = create_diffusion(timestep_respacing=respacing, **config.diffusion)
|
| 88 |
sample_fn = diffusion.ddim_sample_loop_progressive
|
| 89 |
fwd_fn = model.forward_with_cfg
|
|
|
|
| 84 |
output_mat_video_path = os.path.join(config.output_dir, GRADIO_MAT_VIDEO_PATH)
|
| 85 |
output_glb_path = os.path.join(config.output_dir, GRADIO_GLB_PATH)
|
| 86 |
|
| 87 |
+
respacing = "ddim{}".format(input_num_steps)
|
| 88 |
diffusion = create_diffusion(timestep_respacing=respacing, **config.diffusion)
|
| 89 |
sample_fn = diffusion.ddim_sample_loop_progressive
|
| 90 |
fwd_fn = model.forward_with_cfg
|