Update README.md
Browse files
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}
|
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')
|