mirrors/Fooocus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

i

Browse Source
This commit is contained in:
lvmin 2023-08-09 12:55:33 -07:00
parent 1a904ba333
commit b4852a5c0a
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
entry.py
View File

@ -1,5 +1,6 @@
import os import os
import torch import torch
import safetensors.torch
from omegaconf import OmegaConf from omegaconf import OmegaConf
from sgm.util import instantiate_from_config from sgm.util import instantiate_from_config
@ -7,5 +8,9 @@ from sgm.util import instantiate_from_config
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.eval()
sd = safetensors.torch.load_file('./sd_xl_base_1.0.safetensors')
model.load_state_dict(sd, strict=False)
a = 0 a = 0