Skip to content

Commit db9476f

Browse files
authored
Fix docs table (modelscope#1024)
1 parent 9e169a6 commit db9476f

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

swift/llm/utils/model.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2527,12 +2527,12 @@ def get_model_tokenizer_qwen2_chat(model_dir: str,
25272527
function_kwargs={'gptq_bits': 4},
25282528
support_flash_attn=True,
25292529
hf_model_id='Qwen/Qwen1.5-MoE-A2.7B-Chat-GPTQ-Int4')
2530-
def get_model_tokenizer_qwen1half_intx(model_dir: str,
2531-
torch_dtype: Dtype,
2532-
model_kwargs: Dict[str, Any],
2533-
load_model: bool = True,
2534-
**kwargs):
2535-
kwargs['get_qwen_function'] = get_model_tokenizer_qwen1half
2530+
def get_model_tokenizer_qwen2_intx(model_dir: str,
2531+
torch_dtype: Dtype,
2532+
model_kwargs: Dict[str, Any],
2533+
load_model: bool = True,
2534+
**kwargs):
2535+
kwargs['get_qwen_function'] = get_model_tokenizer_qwen2_chat
25362536
return get_model_tokenizer_qwen_intx(model_dir, torch_dtype, model_kwargs, load_model, **kwargs)
25372537

25382538

0 commit comments

Comments
 (0)