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

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

..

cached_model

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

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

__init__.py

Remove ModelCacheBase.

2024-12-04 22:05:34 +00:00

cache_record.py

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

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

cache_stats.py

Move CacheStats to its own file.

2024-12-04 21:56:50 +00:00

model_cache.py

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

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

torch_function_autocast_context.py

Delete experimental torch device autocasting solutions and clean up TorchFunctionAutocastDeviceContext.

2024-12-05 19:36:44 +00:00