Code

Fixes to enable FSDP one-shot#58

Code

Merged

dbogunowicz merged 12 commits intomainneuralmagic/compressed-tensors:mainfrom fsdpneuralmagic/compressed-tensors:fsdpCopy head branch name to clipboardJun 3, 2024

+49-5

Commits

Commits on May 20, 2024

• initial commit

  

  bogunowicz@arrival.com

  committed
  851089d

• check device types when quant

  

  bogunowicz@arrival.com

  committed
  0c5812b

• fix forward

  

  bogunowicz@arrival.com

  committed
  a268e42

Commits on May 22, 2024

• another fix

  

  bogunowicz@arrival.com

  committed
  d70b13b

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

  dbogunowicz

  authored
  88e3f25

Commits on May 23, 2024

• beautify

  

  bogunowicz@arrival.com

  committed
  09cb8b4

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

  

  bogunowicz@arrival.com

  committed
  00e56c6

• attempt fix tests

  

  bogunowicz@arrival.com

  committed
  0e06873

• add warning for undetected ignored modules

  

  bogunowicz@arrival.com

  committed
  0f65266

Commits on May 27, 2024

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

  

  bogunowicz@arrival.com

  committed
  a5f8544

Commits on May 28, 2024

• refactor

  

  bogunowicz@arrival.com

  committed
  8d91989

• Remove the dot

  dbogunowicz

  authored
  8c0203f