Ryan Dick 7214d4969b Workaround a weird quirk of QuantState.to() and add a unit test to exercise it.

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

..

cached_model

Add CachedModelOnlyFullLoad to mirror the CachedModelWithPartialLoad for models that cannot or should not be partially loaded.

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

torch_module_autocast

Workaround a weird quirk of QuantState.to() and add a unit test to exercise it.

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

__init__.py

Remove ModelCacheBase.

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

cache_record.py

Rename model_cache_default.py -> model_cache.py.

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

cache_stats.py

Move CacheStats to its own file.

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

model_cache.py

(minor) Add TODO comment regarding the location of get_model_cache_key().

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