File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -220,11 +220,11 @@ from transformers import (
220
220
AutoTokenizer,
221
221
AutoModelForCausalLM,
222
222
)
223
- tokenizer = AutoTokenizer.from_pretrained(mode_name_or_path, trust_remote_code = True )
224
- tokenizer.padding_side = " left"
225
- tokenizer.eos_token_id = tokenizer.convert_tokens_to_ids(" {EOS_TOKEN} >" )
223
+ model_name_or_path = " codefuse-ai/CodeFuse-Deepseek-33B "
224
+ tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, trust_remote_code = True , padding_side = " left" )
225
+ tokenizer.eos_token_id = tokenizer.convert_tokens_to_ids(" <|end▁of▁sentence| >" )
226
226
tokenizer.pad_token_id = tokenizer.eos_token_id
227
- model = AutoModelForCausalLM.from_pretrained(mode_name_or_path , trust_remote_code = True )
227
+ model = AutoModelForCausalLM.from_pretrained(model_name_or_path , trust_remote_code = True )
228
228
229
229
HUMAN_ROLE_START_TAG = " <s>user\n "
230
230
BOT_ROLE_START_TAG = " <s>assistant\n "
You can’t perform that action at this time.
0 commit comments