Yaron Koresh commited on
Commit
3f070dc
·
verified ·
1 Parent(s): c4dbf25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -30,7 +30,7 @@ height=896
30
  device = "cuda"
31
  dtype = torch.float16
32
  result=[]
33
- step = 8
34
  accu=7.5
35
  repo = "ByteDance/AnimateDiff-Lightning"
36
  ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
@@ -145,7 +145,7 @@ def Piper(image,positive,negative,motion):
145
  pipe.set_adapters(["motion"], [0.7])
146
  last_motion = motion
147
 
148
- pipe.to(device)
149
 
150
  return pipe(
151
  prompt=positive,
 
30
  device = "cuda"
31
  dtype = torch.float16
32
  result=[]
33
+ step = 2
34
  accu=7.5
35
  repo = "ByteDance/AnimateDiff-Lightning"
36
  ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
 
145
  pipe.set_adapters(["motion"], [0.7])
146
  last_motion = motion
147
 
148
+ pipe.to(device,dtype)
149
 
150
  return pipe(
151
  prompt=positive,