Koi953215 commited on
Commit
1d63ac6
1 Parent(s): 79910d2

remove float16

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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", torch_dtype=torch.float16)
205
  pipe = StableDiffusionControlNetPipeline.from_pretrained(
206
- "runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
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", torch_dtype=torch.float16)
232
  pipe = StableDiffusionControlNetPipeline.from_pretrained(
233
- "runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
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