transformers/tests/quantization/autoawq
Younes Belkada 266c67b06a
[Mixtral / Awq] Add mixtral fused modules for Awq (#28240)
* add mixtral fused modules

* add changes from modeling utils

* add test

* fix test + rope theta issue

* Update src/transformers/modeling_utils.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* add tests

---------

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
2024-01-12 14:29:35 +01:00
..
__init__.py
test_awq.py [Mixtral / Awq] Add mixtral fused modules for Awq (#28240) 2024-01-12 14:29:35 +01:00