Ryan Dick 5d41157404 Add LayerNorm to list of modules optimized by skip_torch_weight_init()

2024-04-05 14:39:57 -04: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

Remove RawModel - it was just creating a weird layer of indirection in the AnyModel type without adding any value.

2024-04-04 18:07:13 -04:00

model_loaders

Remove RawModel - it was just creating a weird layer of indirection in the AnyModel type without adding any value.

2024-04-04 18:07:13 -04:00

__init__.py

chore: ruff

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

load_base.py

Remove RawModel - it was just creating a weird layer of indirection in the AnyModel type without adding any value.

2024-04-04 18:07:13 -04:00

load_default.py

Remove RawModel - it was just creating a weird layer of indirection in the AnyModel type without adding any value.

2024-04-04 18:07:13 -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

Remove RawModel - it was just creating a weird layer of indirection in the AnyModel type without adding any value.

2024-04-04 18:07:13 -04:00

optimizations.py

Add LayerNorm to list of modules optimized by skip_torch_weight_init()

2024-04-05 14:39:57 -04:00