Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

i

Browse Source
This commit is contained in:
lvmin
2023-08-09 13:57:29 -07:00
parent ebfe47d02d
commit 573dc98eff
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
entry.py
+1
View File
@@ -91,6 +91,7 @@ sampler = EulerAncestralSampler(
config_path = './sd_xl_base.yaml' config_path = './sd_xl_base.yaml'
config = OmegaConf.load(config_path) config = OmegaConf.load(config_path)
model = instantiate_from_config(config.model).cpu() model = instantiate_from_config(config.model).cpu()
model.to(torch.float16)
model.eval() model.eval()
model.load_state_dict(safetensors.torch.load_file('./sd_xl_base_1.0.safetensors'), strict=False) model.load_state_dict(safetensors.torch.load_file('./sd_xl_base_1.0.safetensors'), strict=False)