From 573dc98eff70267853826d55c39c458791dbfb7d Mon Sep 17 00:00:00 2001 From: lvmin Date: Wed, 9 Aug 2023 13:57:29 -0700 Subject: [PATCH] i --- entry.py | 1 + 1 file changed, 1 insertion(+) diff --git a/entry.py b/entry.py index 9bb1c42a..c0726986 100644 --- a/entry.py +++ b/entry.py @@ -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)