OmPrakashSingh1704 commited on
Commit
f84e193
·
1 Parent(s): bea4d2e
options/Banner_Model/Image2Image_2.py CHANGED
@@ -10,7 +10,7 @@ processor = LineartDetector.from_pretrained("lllyasviel/Annotators")
10
  checkpoint = "ControlNet-1-1-preview/control_v11p_sd15_lineart"
11
  controlnet = ControlNetModel.from_pretrained(checkpoint, torch_dtype=torch.float16).to(device)
12
  pipe = StableVideoDiffusionPipeline.from_pretrained(
13
- "benjamin-paine/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
14
  ).to(device)
15
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
16
  # pipe.enable_model_cpu_offload()
 
10
  checkpoint = "ControlNet-1-1-preview/control_v11p_sd15_lineart"
11
  controlnet = ControlNetModel.from_pretrained(checkpoint, torch_dtype=torch.float16).to(device)
12
  pipe = StableVideoDiffusionPipeline.from_pretrained(
13
+ "stable-diffusion-v1-5/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
14
  ).to(device)
15
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
16
  # pipe.enable_model_cpu_offload()
options/Banner_Model/__pycache__/Image2Image_2.cpython-310.pyc CHANGED
Binary files a/options/Banner_Model/__pycache__/Image2Image_2.cpython-310.pyc and b/options/Banner_Model/__pycache__/Image2Image_2.cpython-310.pyc differ