Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
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(
|