Ryan Dick e5180c4e6b Add get_effective_device(...) utility to aid in determining the effective device of models that are partially loaded.

2025-01-07 00:31:00 +00:00

..

model_cache

Add get_effective_device(...) utility to aid in determining the effective device of models that are partially loaded.

2025-01-07 00:31:00 +00:00

model_loaders

Pull get_model_cache_key(...) out of ModelCache. The ModelCache should not be concerned with implementation details like the submodel_type.

2024-12-24 14:23:18 +00:00

__init__.py

Rename model_cache_default.py -> model_cache.py.

2024-12-24 14:23:18 +00:00

load_base.py

First pass at adding partial loading support to the ModelCache.

2025-01-07 00:30:58 +00:00

load_default.py

Get rid of ModelLocker. It was an unnecessary layer of indirection.

2024-12-24 14:23:18 +00:00

memory_snapshot.py

Add ruff rule to disallow relative parent imports.

2024-07-04 09:35:37 -04:00

model_loader_registry.py

Apply ruff rule to disallow all relative imports.

2024-07-04 09:35:37 -04:00

model_util.py

Rename LoRAModelRaw to ModelPatchRaw.

2024-12-17 13:20:19 +00:00

optimizations.py

final tidying before marking PR as ready for review

2024-03-01 10:42:33 +11:00