Commit fd4bad80 authored by chenpangpang's avatar chenpangpang
Browse files

feat: 解决dcu不支持float16的问题

parent 4e7af75c
...@@ -86,20 +86,20 @@ class LazyLoadPipeline: ...@@ -86,20 +86,20 @@ class LazyLoadPipeline:
def setup_pipeline(self): def setup_pipeline(self):
print("Setting up the pipeline...") print("Setting up the pipeline...")
controlnet = ControlNetModel.from_single_file( controlnet = ControlNetModel.from_single_file(
"models/ControlNet/control_v11f1e_sd15_tile.pth", torch_dtype=torch.float16 "models/ControlNet/control_v11f1e_sd15_tile.pth", torch_dtype=torch.float32
) )
safety_checker = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker") safety_checker = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
model_path = "models/models/Stable-diffusion/juggernaut_reborn.safetensors" model_path = "models/models/Stable-diffusion/juggernaut_reborn.safetensors"
pipe = StableDiffusionControlNetImg2ImgPipeline.from_single_file( pipe = StableDiffusionControlNetImg2ImgPipeline.from_single_file(
model_path, model_path,
controlnet=controlnet, controlnet=controlnet,
torch_dtype=torch.float16, torch_dtype=torch.float32,
use_safetensors=True, use_safetensors=True,
safety_checker=safety_checker safety_checker=safety_checker
) )
vae = AutoencoderKL.from_single_file( vae = AutoencoderKL.from_single_file(
"models/VAE/vae-ft-mse-840000-ema-pruned.safetensors", "models/VAE/vae-ft-mse-840000-ema-pruned.safetensors",
torch_dtype=torch.float16 torch_dtype=torch.float32
) )
pipe.vae = vae pipe.vae = vae
pipe.load_textual_inversion("models/embeddings/verybadimagenegative_v1.3.pt") pipe.load_textual_inversion("models/embeddings/verybadimagenegative_v1.3.pt")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment