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 |
||
|---|---|---|
| .. | ||
| image_util | ||
| ip_adapter | ||
| model_hash | ||
| model_manager | ||
| onnx | ||
| stable_diffusion | ||
| tiles | ||
| util | ||
| __init__.py | ||
| lora.py | ||
| model_patcher.py | ||
| raw_model.py | ||
| textual_inversion.py | ||