mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
243de683d7
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 |
||
---|---|---|
.. | ||
load | ||
metadata | ||
util | ||
__init__.py | ||
config.py | ||
convert_ckpt_to_diffusers.py | ||
libc_util.py | ||
merge.py | ||
probe.py | ||
search.py | ||
starter_models.py |