Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -121,14 +121,14 @@ adapter = MotionAdapter.from_pretrained(motion_adapter, torch_dtype=dtype, safet
|
|
121 |
controlnet = SparseControlNetModel.from_pretrained(controlnet_id, torch_dtype=torch.float16).to(device)
|
122 |
vae = AutoencoderKL.from_pretrained(vae_id, torch_dtype=torch.float16).to(device)
|
123 |
scheduler = DPMSolverMultistepScheduler.from_pretrained(
|
124 |
-
|
125 |
subfolder="scheduler",
|
126 |
beta_schedule="linear",
|
127 |
algorithm_type="dpmsolver++",
|
128 |
use_karras_sigmas=True,
|
129 |
)
|
130 |
pipe = AnimateDiffSparseControlNetPipeline.from_pretrained(
|
131 |
-
|
132 |
motion_adapter=motion_adapter,
|
133 |
controlnet=controlnet,
|
134 |
vae=vae,
|
|
|
121 |
controlnet = SparseControlNetModel.from_pretrained(controlnet_id, torch_dtype=torch.float16).to(device)
|
122 |
vae = AutoencoderKL.from_pretrained(vae_id, torch_dtype=torch.float16).to(device)
|
123 |
scheduler = DPMSolverMultistepScheduler.from_pretrained(
|
124 |
+
base,
|
125 |
subfolder="scheduler",
|
126 |
beta_schedule="linear",
|
127 |
algorithm_type="dpmsolver++",
|
128 |
use_karras_sigmas=True,
|
129 |
)
|
130 |
pipe = AnimateDiffSparseControlNetPipeline.from_pretrained(
|
131 |
+
base,
|
132 |
motion_adapter=motion_adapter,
|
133 |
controlnet=controlnet,
|
134 |
vae=vae,
|