transformers/docs/source/en
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
..
internal
main_classes HIGGS Quantization Support (#34997) 2024-12-23 16:54:49 +01:00
model_doc
quantization HIGGS Quantization Support (#34997) 2024-12-23 16:54:49 +01:00
tasks
_config.py
_redirects.yml
_toctree.yml HIGGS Quantization Support (#34997) 2024-12-23 16:54:49 +01:00
accelerate.md
add_new_model.md
add_new_pipeline.md
agents.md
agents_advanced.md
attention.md
autoclass_tutorial.md
benchmarks.md
bertology.md
big_models.md
chat_templating.md
community.md
contributing.md
conversations.md
create_a_model.md
custom_models.md
debugging.md
deepspeed.md
fast_tokenizers.md
fsdp.md
generation_strategies.md
gguf.md
glossary.md
how_to_hack_models.md
hpo_train.md
index.md
installation.md
kv_cache.md
llm_optims.md
llm_tutorial.md
llm_tutorial_optimization.md
model_memory_anatomy.md
model_sharing.md
model_summary.md
modular_transformers.md
multilingual.md
notebooks.md
pad_truncation.md
peft.md
perf_hardware.md
perf_infer_cpu.md
perf_infer_gpu_multi.md
perf_infer_gpu_one.md
perf_torch_compile.md
perf_train_cpu.md
perf_train_cpu_many.md
perf_train_gpu_many.md
perf_train_gpu_one.md
perf_train_special.md
perf_train_tpu_tf.md
performance.md
perplexity.md
philosophy.md
pipeline_tutorial.md
pipeline_webserver.md
pr_checks.md
preprocessing.md
quicktour.md
run_scripts.md
sagemaker.md
serialization.md
task_summary.md
tasks_explained.md
testing.md
tf_xla.md
tflite.md
tiktoken.md
tokenizer_summary.md
torchscript.md
trainer.md
training.md
troubleshooting.md