Yaron Koresh commited on
Commit
72c1344
·
verified ·
1 Parent(s): 4513e12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -116,7 +116,7 @@ function custom(){
116
 
117
  # torch pipe
118
 
119
- pipe = AnimateDiffPipeline.from_pretrained(base, motion_adapter=adapter, torch_dtype=dtype).to(device)
120
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_pretrained(
121
  base,
122
  subfolder="scheduler",
@@ -315,7 +315,7 @@ def calc(img,p1,p2,motion):
315
  pipe.set_adapters("motion", [0.7])
316
  last_motion = motion
317
 
318
- pipe.to(device,dtype)
319
 
320
  return pipe(
321
  prompt=p1,
 
116
 
117
  # torch pipe
118
 
119
+ pipe = AnimateDiffPipeline.from_pretrained(base, motion_adapter=adapter).to(device)
120
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_pretrained(
121
  base,
122
  subfolder="scheduler",
 
315
  pipe.set_adapters("motion", [0.7])
316
  last_motion = motion
317
 
318
+ pipe.to(device)
319
 
320
  return pipe(
321
  prompt=p1,