no-half support for SD 2.0

3 years ago · 1e506657e1
parent b5050ad207
commit 1e506657e1
1 changed files with 3 additions and 0 deletions
--- 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)