From 05b01da01f76c97011358a415820360546d284fb Mon Sep 17 00:00:00 2001 From: catboxanon <122327233+catboxanon@users.noreply.github.com> Date: Sat, 2 Nov 2024 10:29:13 -0400 Subject: [PATCH] Fix ldm .yaml loading typo (#2248) --- backend/loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/loader.py b/backend/loader.py index fa157ac2..5eff980b 100644 --- a/backend/loader.py +++ b/backend/loader.py @@ -315,7 +315,7 @@ def forge_loader(sd, additional_state_dicts=None): if yaml_config is not None: yaml_config_prediction_type: str = ( yaml_config.get('model', {}).get('params', {}).get('parameterization', '') - or yaml_config.get('model', {}).get('params', {}).get('denoiser_config', {}).get('params', {}).get('scaling_config').get('target', '') + or yaml_config.get('model', {}).get('params', {}).get('denoiser_config', {}).get('params', {}).get('scaling_config', {}).get('target', '') ) if yaml_config_prediction_type == 'v' or yaml_config_prediction_type.endswith(".VScaling"): yaml_config_prediction_type = 'v_prediction'