Spaces:
Running
Running
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -48,7 +48,7 @@ else:
|
|
48 |
dtype = torch.float16
|
49 |
|
50 |
base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
|
51 |
-
adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device)
|
52 |
|
53 |
# variable data
|
54 |
|
@@ -109,7 +109,7 @@ function custom(){
|
|
109 |
# torch pipes
|
110 |
|
111 |
image_pipe = StableDiffusionPipeline.from_pretrained(base, torch_dtype=dtype, safety_checker=None).to(device)
|
112 |
-
pipe = AnimateDiffPipeline.from_pretrained(base, torch_dtype=dtype, motion_adapter=adapter).to(device)
|
113 |
pipe.scheduler = EulerAncestralDiscreteScheduler(
|
114 |
base,
|
115 |
subfolder="scheduler",
|
|
|
48 |
dtype = torch.float16
|
49 |
|
50 |
base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
|
51 |
+
adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device, safety_checker=None)
|
52 |
|
53 |
# variable data
|
54 |
|
|
|
109 |
# torch pipes
|
110 |
|
111 |
image_pipe = StableDiffusionPipeline.from_pretrained(base, torch_dtype=dtype, safety_checker=None).to(device)
|
112 |
+
pipe = AnimateDiffPipeline.from_pretrained(base, torch_dtype=dtype, motion_adapter=adapter, safety_checker=None).to(device)
|
113 |
pipe.scheduler = EulerAncestralDiscreteScheduler(
|
114 |
base,
|
115 |
subfolder="scheduler",
|