michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

no-half support for SD 2.0

Browse source
This commit is contained in:
MrCheeze 2022-11-26 13:28:44 -05:00
parent b5050ad207
commit 1e506657e1
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
modules/sd_models.py
View file

@ -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)