Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix model variant

Browse Source
This commit is contained in:
Vladimir Mandic
2024-01-27 09:10:21 -05:00
parent ffee8d65af
commit 389f86dc5f
1 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models.py
+1 -7
View File
@@ -740,14 +740,8 @@ def load_diffuser(checkpoint_info=None, already_loaded_state_dict=None, timer=No
"load_safety_checker": False,
"load_connected_pipeline": True,
}
if shared.opts.diffusers_model_load_variant == 'default':
if devices.dtype == torch.float16:
diffusers_load_config['variant'] = 'fp16'
elif shared.opts.diffusers_model_load_variant == 'fp32':
pass
else:
if shared.opts.diffusers_model_load_variant != 'default':
diffusers_load_config['variant'] = shared.opts.diffusers_model_load_variant
if shared.opts.diffusers_pipeline == 'Custom Diffusers Pipeline' and len(shared.opts.custom_diffusers_pipeline) > 0:
shared.log.debug(f'Diffusers custom pipeline: {shared.opts.custom_diffusers_pipeline}')
diffusers_load_config['custom_pipeline'] = shared.opts.custom_diffusers_pipeline