InvokeAI/invokeai/backend/model_manager
psychedelicious 243de683d7 fix(mm): handle integer state dict keys in probe
It's possible for a model's state dict to have integer keys, though we do not actually support such models.

As part of probing, we call `key.startswith(...)` on the state dict keys. This raises an `AttributeError` for integer keys.

This logic is in `invokeai/backend/model_manager/probe.py:get_model_type_from_checkpoint`

To fix this, we can cast the keys to strings first. The models w/ integer keys will still fail to be probed, but we'll get a `InvalidModelConfigException` instead of `AttributeError`.

Closes #6044
2024-03-27 09:30:25 +11:00
..
load allow checkpoint config files to use root-relative paths 2024-03-22 08:57:45 +11:00
metadata fix(mm): config.json to indicates diffusers model 2024-03-13 21:02:29 +11:00
util install model if diffusers or single file, cleaned up backend logic to not mess with existing model install 2024-03-13 21:02:29 +11:00
__init__.py chore: ruff 2024-03-01 10:42:33 +11:00
config.py fix(mm): default settings pydantic error 2024-03-25 07:40:52 -04:00
convert_ckpt_to_diffusers.py Remove core safetensors->diffusers conversion models 2024-03-17 19:13:18 -04:00
libc_util.py Tidy names and locations of modules 2024-03-01 10:42:33 +11:00
merge.py fix(config): use new get_config across the app, use correct settings 2024-03-19 09:24:28 +11:00
probe.py fix(mm): handle integer state dict keys in probe 2024-03-27 09:30:25 +11:00
search.py docs(mm): format docstrings for ModelSearch 2024-03-10 12:09:47 +11:00
starter_models.py fix(mm): remove proteus model 2024-03-22 02:22:03 -07:00