Yaron Koresh commited on
Commit
8a6d9b6
·
verified ·
1 Parent(s): 08ee281

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -125,7 +125,7 @@ scheduler = DPMSolverMultistepScheduler.from_pretrained(
125
  subfolder="scheduler",
126
  beta_schedule="linear",
127
  algorithm_type="dpmsolver++",
128
- use_karras_sigmas=True,
129
  )
130
  pipe = AnimateDiffSparseControlNetPipeline.from_pretrained(
131
  base,
@@ -133,7 +133,7 @@ pipe = AnimateDiffSparseControlNetPipeline.from_pretrained(
133
  controlnet=controlnet,
134
  vae=vae,
135
  scheduler=scheduler,
136
- torch_dtype=torch.float16,
137
  ).to(device)
138
 
139
  pipe.scheduler = DDIMScheduler.from_pretrained(
 
125
  subfolder="scheduler",
126
  beta_schedule="linear",
127
  algorithm_type="dpmsolver++",
128
+ use_karras_sigmas=True
129
  )
130
  pipe = AnimateDiffSparseControlNetPipeline.from_pretrained(
131
  base,
 
133
  controlnet=controlnet,
134
  vae=vae,
135
  scheduler=scheduler,
136
+ torch_dtype=torch.float16
137
  ).to(device)
138
 
139
  pipe.scheduler = DDIMScheduler.from_pretrained(