Merge pull request #5085 from MrCheeze/sd-2.0-automatic-2

no-half support for SD 2.0
This commit is contained in:
AUTOMATIC1111 2022-11-27 13:54:08 +03:00 committed by GitHub
commit 8de897b3da
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions

View File

@ -244,6 +244,9 @@ def load_model(checkpoint_info=None):
do_inpainting_hijack() 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) sd_model = instantiate_from_config(sd_config.model)
load_model_weights(sd_model, checkpoint_info) load_model_weights(sd_model, checkpoint_info)