diff options
author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-11-27 10:54:08 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-11-27 10:54:08 +0000 |
commit | 8de897b3daf4cf8ca928546e052c2e2f38ad6e01 (patch) | |
tree | eb9b967e09cc0c131fb851bddcb2755c03e34f83 /modules/sd_models.py | |
parent | d46d73c3c2e9b1641ffb3fa34be6448828e3f863 (diff) | |
parent | 1e506657e1cb732a5f0e567ba2585fba2bbb1327 (diff) | |
download | stable-diffusion-webui-gfx803-8de897b3daf4cf8ca928546e052c2e2f38ad6e01.tar.gz stable-diffusion-webui-gfx803-8de897b3daf4cf8ca928546e052c2e2f38ad6e01.tar.bz2 stable-diffusion-webui-gfx803-8de897b3daf4cf8ca928546e052c2e2f38ad6e01.zip |
Merge pull request #5085 from MrCheeze/sd-2.0-automatic-2
no-half support for SD 2.0
Diffstat (limited to 'modules/sd_models.py')
-rw-r--r-- | modules/sd_models.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index c59151e0..0e0bd79e 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -244,6 +244,9 @@ def load_model(checkpoint_info=None): do_inpainting_hijack()
+ if shared.cmd_opts.no_half:
+ sd_config.model.params.unet_config.params.use_fp16 = False
+
sd_model = instantiate_from_config(sd_config.model)
load_model_weights(sd_model, checkpoint_info)
|