i

2026-08-16 13:13:16 +02:00 · 2023-08-09 13:57:29 -07:00
parent ebfe47d02d
commit 573dc98eff
1 changed files with 1 additions and 0 deletions
@@ -91,6 +91,7 @@ sampler = EulerAncestralSampler(
 config_path = './sd_xl_base.yaml'
 config = OmegaConf.load(config_path)
 model = instantiate_from_config(config.model).cpu()
+model.to(torch.float16)
 model.eval()
 model.load_state_dict(safetensors.torch.load_file('./sd_xl_base_1.0.safetensors'), strict=False)