codefuse-admin commited on
Commit
3dbd540
1 Parent(s): bc79cfb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -149,7 +149,7 @@ HUMAN_ROLE_START_TAG = "<s>human\n"
149
  BOT_ROLE_START_TAG = "<s>bot\n"
150
 
151
  model_dir = snapshot_download('codefuse-ai/CodeFuse-DeepSeek-33B', revision='v1.0.0')
152
- text_list = [f'{HUMAN_ROLE_START_TAG}请写一个快排程序\n#Python\n{BOT_ROLE_START_TAG}']
153
 
154
  model, tokenizer = load_model_tokenizer(model_dir)
155
  inputs = tokenizer(text_list, return_tensors='pt', padding=True, add_special_tokens=False).to('cuda')
 
149
  BOT_ROLE_START_TAG = "<s>bot\n"
150
 
151
  model_dir = snapshot_download('codefuse-ai/CodeFuse-DeepSeek-33B', revision='v1.0.0')
152
+ text_list = [f'{HUMAN_ROLE_START_TAG}Write a QuickSort program\n#Python\n{BOT_ROLE_START_TAG}']
153
 
154
  model, tokenizer = load_model_tokenizer(model_dir)
155
  inputs = tokenizer(text_list, return_tensors='pt', padding=True, add_special_tokens=False).to('cuda')