Spaces:
Running
Running
Commit
·
30f2818
1
Parent(s):
f84e193
update
Browse files
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
|
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
|
|