Yaron Koresh commited on
Commit
f031971
·
verified ·
1 Parent(s): 13c45fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -136,9 +136,9 @@ scheduler2 = DDIMScheduler.from_pretrained(
136
 
137
  pipe = AnimateDiffPipeline.from_pretrained(
138
  base,
139
- motion_adapter=motion_adapter,
140
  torch_dtype=torch.float16
141
- )
142
 
143
  pipe.scheduler = scheduler2
144
  pipe.vae = vae
 
136
 
137
  pipe = AnimateDiffPipeline.from_pretrained(
138
  base,
139
+ motion_adapter=adapter,
140
  torch_dtype=torch.float16
141
+ ).to(device)
142
 
143
  pipe.scheduler = scheduler2
144
  pipe.vae = vae