Spaces:
Running
Running
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -116,7 +116,7 @@ function custom(){
|
|
116 |
|
117 |
# torch pipe
|
118 |
|
119 |
-
pipe = AnimateDiffPipeline.from_pretrained(base, motion_adapter=adapter
|
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
|
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,
|