Update src/transformers/quantizers/quantizer_quark.py

Co-authored-by: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com>
This commit is contained in:
amd-xiaoyu12 2025-04-28 20:17:22 -04:00 committed by GitHub
parent 300358ebfa
commit f20c40f0f1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -119,7 +119,7 @@ class QuarkHfQuantizer(HfQuantizer):
if config.get_text_config() is not None:
text_plan = config.get_text_config().base_model_tp_plan
# modify exsiting play
# modify existing play
text_plan_cp = {}
for key in text_plan:
text_plan_cp[key+".weight"] = text_plan[key]