psychedelicious 59b4a23479 feat(mm): use same pattern for vae converter as others ...

Add `dump_path` arg to the converter function & save the model to disk inside the conversion function. This is the same pattern as in the other conversion functions.

2024-04-01 12:34:49 +11:00

..

convert_cache

[mm] Do not write diffuser model to disk when convert_cache set to zero (#6072)

2024-03-29 16:11:08 -04:00

model_cache

[mm] Do not write diffuser model to disk when convert_cache set to zero (#6072)

2024-03-29 16:11:08 -04:00

model_loaders

feat(mm): use same pattern for vae converter as others

2024-04-01 12:34:49 +11:00

__init__.py

chore: ruff

2024-03-01 10:42:33 +11:00

load_base.py

[mm] Do not write diffuser model to disk when convert_cache set to zero (#6072)

2024-03-29 16:11:08 -04:00

load_default.py

[mm] Do not write diffuser model to disk when convert_cache set to zero (#6072)

2024-03-29 16:11:08 -04:00

memory_snapshot.py

final tidying before marking PR as ready for review

2024-03-01 10:42:33 +11:00

model_loader_registry.py

Experiment with using absolute paths within model management

2024-03-08 15:36:14 -05:00

model_util.py

make model manager v2 ready for PR review

2024-03-01 10:42:33 +11:00

optimizations.py

final tidying before marking PR as ready for review

2024-03-01 10:42:33 +11:00