Ryan Dick c7b84cf012 WIP - first pass at overhauling ModelCache to work with partial loads.

2024-12-05 23:03:40 +00:00

..

model_cache

WIP - first pass at overhauling ModelCache to work with partial loads.

2024-12-05 23:03:40 +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-04 22:53:57 +00:00

__init__.py

Rename model_cache_default.py -> model_cache.py.

2024-12-04 22:45:30 +00:00

load_base.py

WIP - first pass at overhauling ModelCache to work with partial loads.

2024-12-05 23:03:40 +00:00

load_default.py

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

2024-12-05 16:59:40 +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

Add util functions calc_tensor_size(...) and calc_tensors_size(...).

2024-09-15 04:39:56 +03:00

optimizations.py

final tidying before marking PR as ready for review

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