Guangyao Zhang e96a0761ea [FP8] unsqueeze scale to make it compatible with torch.compile (#6040)

2024-08-29 14:49:23 +08:00

..

__init__.py

[Feature] qlora support (#5586)

2024-04-28 10:51:27 +08:00

bnb_config.py

[Feature] qlora support (#5586)

2024-04-28 10:51:27 +08:00

bnb.py

[quant] fix bitsandbytes version check (#5882)

2024-07-04 11:33:23 +08:00

fp8_hook.py

[fp8] support gemini plugin (#5978)

2024-08-09 14:09:48 +08:00

fp8.py

[FP8] unsqueeze scale to make it compatible with torch.compile (#6040)

2024-08-29 14:49:23 +08:00

utils.py

[Feature]: support FP8 communication in DDP, FSDP, Gemini (#5928)

2024-08-08 15:55:01 +08:00