Spaces:
Running
on
Zero
Running
on
Zero
remove float16
Browse files
app.py
CHANGED
@@ -201,9 +201,9 @@ def edit_with_pnp(input_video, prompt, num_steps, guidance_scale, seed, n_prompt
|
|
201 |
|
202 |
if control_type == "Lineart":
|
203 |
# Load the control net model for lineart
|
204 |
-
controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_lineart"
|
205 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
206 |
-
"runwayml/stable-diffusion-v1-5", controlnet=controlnet
|
207 |
)
|
208 |
pipe.to(device)
|
209 |
# lineart
|
@@ -228,9 +228,9 @@ def edit_with_pnp(input_video, prompt, num_steps, guidance_scale, seed, n_prompt
|
|
228 |
|
229 |
else:
|
230 |
# Load the control net model for canny
|
231 |
-
controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_canny"
|
232 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
233 |
-
"runwayml/stable-diffusion-v1-5", controlnet=controlnet
|
234 |
)
|
235 |
pipe.to(device)
|
236 |
# canny
|
|
|
201 |
|
202 |
if control_type == "Lineart":
|
203 |
# Load the control net model for lineart
|
204 |
+
controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_lineart")
|
205 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
206 |
+
"runwayml/stable-diffusion-v1-5", controlnet=controlnet
|
207 |
)
|
208 |
pipe.to(device)
|
209 |
# lineart
|
|
|
228 |
|
229 |
else:
|
230 |
# Load the control net model for canny
|
231 |
+
controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_canny")
|
232 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
233 |
+
"runwayml/stable-diffusion-v1-5", controlnet=controlnet
|
234 |
)
|
235 |
pipe.to(device)
|
236 |
# canny
|