Ryan Dick fe0ef2c27c Add torch module autocast utilities.

2024-12-24 14:32:11 +00:00

..

model_cache

Add torch module autocast utilities.

2024-12-24 14:32:11 +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

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

2024-12-24 14:23:18 +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