Ryan Dick 987c9ae076 Move custom autocast modules to separate files in a custom_modules/ directory.

2024-12-24 22:21:31 +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

Move custom autocast modules to separate files in a custom_modules/ directory.

2024-12-24 22:21:31 +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