David Burnett 54b3aa1d01 load t5 model in the same format as it is saved, seems to load as float32 on Macs

2024-10-23 10:09:24 +11:00

..

model_cache

[MM] add API routes for getting & setting MM cache sizes (#6523)

2024-09-02 12:18:21 -04:00

model_loaders

load t5 model in the same format as it is saved, seems to load as float32 on Macs

2024-10-23 10:09:24 +11:00

__init__.py

Apply ruff rule to disallow all relative imports.

2024-07-04 09:35:37 -04:00

load_base.py

Load single-file checkpoints directly without conversion (#6510)

2024-06-27 17:31:28 -04:00

load_default.py

[MM] add API routes for getting & setting MM cache sizes (#6523)

2024-09-02 12:18:21 -04: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