SahaniJi commited on
Commit
9f19502
1 Parent(s): b8dc5fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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"