mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
Update src/transformers/quantizers/quantizer_quark.py
Co-authored-by: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com>
This commit is contained in:
parent
300358ebfa
commit
f20c40f0f1
@ -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]
|
||||
|
Loading…
Reference in New Issue
Block a user