Spaces:

OmPrakashSingh1704
/

ADVERTISE

Running

App Files Files Community

OmPrakashSingh1704 commited on Sep 17, 2024

Commit

94a2d08

1 Parent(s): 30f2818

update

Browse files

Files changed (2) hide show

options/Banner_Model/Image2Image_2.py +2 -2
options/Banner_Model/__pycache__/Image2Image_2.cpython-310.pyc +0 -0

options/Banner_Model/Image2Image_2.py CHANGED Viewed

@@ -1,6 +1,6 @@
 import torch
 from controlnet_aux import LineartDetector
-from diffusers import ControlNetModel,UniPCMultistepScheduler,StableVideoDiffusionPipeline
 from PIL import Image
 device= "cuda" if torch.cuda.is_available() else "cpu"
@@ -9,7 +9,7 @@ processor = LineartDetector.from_pretrained("lllyasviel/Annotators")
 checkpoint = "ControlNet-1-1-preview/control_v11p_sd15_lineart"
 controlnet = ControlNetModel.from_pretrained(checkpoint, torch_dtype=torch.float16).to(device)
-pipe = StableVideoDiffusionPipeline.from_pretrained(
     "radames/stable-diffusion-v1-5-img2img", controlnet=controlnet, torch_dtype=torch.float16
 ).to(device)
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)

 import torch
 from controlnet_aux import LineartDetector
+from diffusers import ControlNetModel,UniPCMultistepScheduler,StableDiffusionControlNetPipeline
 from PIL import Image
 device= "cuda" if torch.cuda.is_available() else "cpu"
 checkpoint = "ControlNet-1-1-preview/control_v11p_sd15_lineart"
 controlnet = ControlNetModel.from_pretrained(checkpoint, torch_dtype=torch.float16).to(device)
+pipe = StableDiffusionControlNetPipeline.from_pretrained(
     "radames/stable-diffusion-v1-5-img2img", controlnet=controlnet, torch_dtype=torch.float16
 ).to(device)
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)

options/Banner_Model/__pycache__/Image2Image_2.cpython-310.pyc CHANGED Viewed

Binary files a/options/Banner_Model/__pycache__/Image2Image_2.cpython-310.pyc and b/options/Banner_Model/__pycache__/Image2Image_2.cpython-310.pyc differ