SparseML Compression Pt 2: Load compressed weights#2184

Merged

mgoin merged 41 commits intomainfrom tensor_decompression

Mar 20, 2024

+372-54

Commits on Mar 12, 2024

initial classes
Sara Adkins
committed
WIP
Sara Adkins
committed
compression working
Sara Adkins
committed
unit tests and README
Sara Adkins
committed
docstrings
Sara Adkins
committed
README and fix test
Sara Adkins
committed
add bitmask source
Sara Adkins
committed
Merge branch 'main' into tensor_compression
Sara Adkins
authored

Commits on Mar 14, 2024

initial commit
Sara Adkins
committed
compression working
Sara Adkins
committed
formatting
Sara Adkins
committed

Commits on Mar 15, 2024

cleanup
Sara Adkins
committed
dtype tests
Sara Adkins
committed
Merge branch 'main' into tensor_compression
Sara Adkins
authored
oops fix test
Sara Adkins
committed
Merge branch 'tensor_compression' of github.com:neuralmagic/sparseml into tensor_compression
Sara Adkins
committed
tests
Sara Adkins
committed
add bfloat16
Sara Adkins
committed
Merge branch 'tensor_compression' into tensor_decompression
Sara Adkins
committed
unit tests
Sara Adkins
committed
docstrings
Sara Adkins
committed
update README
Sara Adkins
committed

Commits on Mar 18, 2024

move statements to debug
Sara Adkins
committed

Commits on Mar 19, 2024

warn on conflicts, store device
Sara Adkins
committed
Merge branch 'main' into tensor_compression
Sara Adkins
authored
Merge branch 'tensor_compression' into tensor_decompression
Sara Adkins
committed
merge conflict
Sara Adkins
committed
fix typing
Sara Adkins
committed
helper fn for setting layers
Sara Adkins
committed
expand cuda memory helper
Sara Adkins
committed
update example and tests
Sara Adkins
committed
update docstrings
Sara Adkins
committed

Commits on Mar 20, 2024

remove unneeded file
Sara Adkins
committed
Merge branch 'tensor_compression' into tensor_decompression
Sara Adkins
committed
Update src/sparseml/pytorch/model_load/helpers.py

Sara Adkins
and
dbogunowicz
authored
Update README.md
Sara Adkins
authored
Merge branch 'main' into tensor_compression
Sara Adkins
authored
Merge branch 'tensor_compression' into tensor_decompression
Sara Adkins
committed
Merge branch 'main' into tensor_decompression
Sara Adkins
committed
Merge branch 'main' into tensor_decompression
Sara Adkins
authored
fix merge conflict error
Sara Adkins
committed