Fixes to enable FSDP one-shot #58

dbogunowicz · 2024-05-20T10:01:53Z

No description provided.

src/compressed_tensors/quantization/lifecycle/apply.py

src/compressed_tensors/utils/helpers.py

bogunowicz@arrival.com added 4 commits May 20, 2024 10:01

initial commit

851089d

check device types when quant

0c5812b

fix forward

a268e42

another fix

d70b13b

dbogunowicz commented May 22, 2024

View reviewed changes

src/compressed_tensors/quantization/lifecycle/apply.py Outdated Show resolved Hide resolved

dbogunowicz and others added 5 commits May 22, 2024 10:32

Update src/compressed_tensors/quantization/lifecycle/apply.py

88e3f25

beautify

09cb8b4

erge remote-tracking branch 'origin/main' into fsdp

00e56c6

attempt fix tests

0e06873

add warning for undetected ignored modules

0f65266

dbogunowicz marked this pull request as ready for review May 24, 2024 11:28

Merge remote-tracking branch 'origin/main' into fsdp

a5f8544

dbogunowicz mentioned this pull request May 28, 2024

[Fix] Fully functional FSDP one-shot process neuralmagic/sparseml#2305

Merged

refactor

8d91989

dbogunowicz commented May 28, 2024

View reviewed changes

src/compressed_tensors/utils/helpers.py Outdated Show resolved Hide resolved

Remove the dot

8c0203f

Satrat approved these changes May 28, 2024

View reviewed changes

dbogunowicz requested a review from bfineran May 28, 2024 17:22

bfineran approved these changes Jun 3, 2024

View reviewed changes

dbogunowicz merged commit 3c8c5ab into main Jun 3, 2024
1 check passed

dbogunowicz deleted the fsdp branch June 3, 2024 11:39

Provide feedback