OmPrakashSingh1704 commited on
Commit
30f2818
·
1 Parent(s): f84e193
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
- "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()
 
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
+ "radames/stable-diffusion-v1-5-img2img", 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