Yaron Koresh commited on
Commit
08ee281
·
verified ·
1 Parent(s): 7474c0a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- model_id,
125
  subfolder="scheduler",
126
  beta_schedule="linear",
127
  algorithm_type="dpmsolver++",
128
  use_karras_sigmas=True,
129
  )
130
  pipe = AnimateDiffSparseControlNetPipeline.from_pretrained(
131
- model_id,
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,