Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -48,6 +48,8 @@ def generate_image(prompt, base, motion, step, progress=gr.Progress()):
|
|
48 |
global motion_loaded
|
49 |
print(prompt, base, step)
|
50 |
|
|
|
|
|
51 |
if step_loaded != step:
|
52 |
repo = "ByteDance/AnimateDiff-Lightning"
|
53 |
ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
|
|
|
48 |
global motion_loaded
|
49 |
print(prompt, base, step)
|
50 |
|
51 |
+
step = int(step)
|
52 |
+
|
53 |
if step_loaded != step:
|
54 |
repo = "ByteDance/AnimateDiff-Lightning"
|
55 |
ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
|