Update README.md

README.md

@@ -165,14 +165,13 @@ def load_model_tokenizer(model_path):
     tokenizer.pad_token_id = tokenizer.convert_tokens_to_ids(tokenizer.pad_token)
     tokenizer.padding_side = "left"
     
-    model = AutoModelForCausalLM.from_pretrained(model_path, device_map='auto',torch_dtype=torch.bfloat16, trust_remote_code=True)
+    model = AutoModelForCausalLM.from_pretrained('codefuse-ai/CodeFuse-DeepSeek-33B', device_map='auto',torch_dtype=torch.bfloat16, trust_remote_code=True)
     return model, tokenizer
 
 
 HUMAN_ROLE_START_TAG = "<s>human\n"
 BOT_ROLE_START_TAG = "<s>bot\n"
 
-model_dir = snapshot_download('codefuse-ai/CodeFuse-DeepSeek-33B', revision='v1.0.0')
 text_list = [f'{HUMAN_ROLE_START_TAG}Write a QuickSort program\n#Python\n{BOT_ROLE_START_TAG}']
 
 model, tokenizer = load_model_tokenizer(model_dir)
@@ -351,14 +350,13 @@ def load_model_tokenizer(model_path):
     tokenizer.pad_token_id = tokenizer.convert_tokens_to_ids(tokenizer.pad_token)
     tokenizer.padding_side = "left"
     
-    model = AutoModelForCausalLM.from_pretrained(model_path, device_map='auto',torch_dtype=torch.bfloat16, trust_remote_code=True)
+    model = AutoModelForCausalLM.from_pretrained('codefuse-ai/CodeFuse-DeepSeek-33B', device_map='auto',torch_dtype=torch.bfloat16, trust_remote_code=True)
     return model, tokenizer
 
 
 HUMAN_ROLE_START_TAG = "<s>human\n"
 BOT_ROLE_START_TAG = "<s>bot\n"
 
-model_dir = snapshot_download('codefuse-ai/CodeFuse-DeepSeek-33B', revision='v1.0.0')
 text_list = [f'{HUMAN_ROLE_START_TAG}请写一个快排程序\n#Python\n{BOT_ROLE_START_TAG}']
 
 model, tokenizer = load_model_tokenizer(model_dir)