Andrei Panferov
64c05eecd6
HIGGS Quantization Support (#34997)
* higgs init
* working with crunches
* per-model workspaces
* style
* style 2
* tests and style
* higgs tests passing
* protecting torch import
* removed torch.Tensor type annotations
* torch.nn.Module inheritance fix maybe
* hide inputs inside quantizer calls
* style structure something
* Update src/transformers/quantizers/quantizer_higgs.py
Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>
* reworked num_sms
* Update src/transformers/integrations/higgs.py
Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>
* revamped device checks
* docstring upd
* Update src/transformers/quantizers/quantizer_higgs.py
Co-authored-by: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com>
* edited tests and device map assertions
* minor edits
* updated flute cuda version in docker
* Added p=1 and 2,3bit HIGGS
* flute version check update
* incorporated `modules_to_not_convert`
* less hardcoding
* Fixed comment
* Added docs
* Fixed gemma support
* example in docs
* fixed torch_dtype for HIGGS
* Update docs/source/en/quantization/higgs.md
Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>
* Collection link
* dequantize interface
* newer flute version, torch.compile support
* unittest message fix
* docs update compile
* isort
* ValueError instead of assert
---------
Co-authored-by: Marc Sun <57196510+SunMarc@users.noreply.github.com>
Co-authored-by: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com>
2024-12-23 16:54:49 +01:00
..
2024-11-26 11:09:30 +01:00
2024-10-04 16:25:10 +02:00
2024-11-25 16:47:14 +01:00
2024-12-18 09:49:59 -05:00
2024-12-13 08:23:31 +01:00
2024-12-09 14:13:36 +01:00
2024-09-11 13:26:44 +02:00
2024-12-16 13:21:44 +01:00
2024-06-17 17:56:51 +01:00
2024-12-23 16:54:49 +01:00
2024-09-30 14:47:18 +02:00
2024-10-02 15:14:34 +02:00
2024-11-20 20:28:51 +01:00
2024-12-23 16:27:46 +01:00