lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-06 21:49:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #2117 from lllyasviel/fix/prediction-detect

Browse Source
This commit is contained in:
catboxanon
2024-10-19 07:48:45 -04:00
committed by GitHub
parent 0511318eff f620f55e56
commit 534405e597
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/loader.py
View File

@@ -294,7 +294,7 @@ def forge_loader(sd, additional_state_dicts=None):
'EDM': 'edm',
}
if 'scheduler' in huggingface_components and hasattr(huggingface_components['scheduler'], 'config') and 'prediction_type' in huggingface_components['scheduler'].config:
huggingface_components['scheduler'].config.prediction_type = prediction_types.get(estimated_config.model_type.name, 'epsilon')
huggingface_components['scheduler'].config.prediction_type = prediction_types.get(estimated_config.model_type.name, huggingface_components['scheduler'].config.prediction_type)
for M in possible_models:
if any(isinstance(estimated_config, x) for x in M.matched_guesses):