Update README.md
Browse files
README.md
CHANGED
@@ -149,7 +149,7 @@ for data in tqdm(datasets):
|
|
149 |
|
150 |
prompt = f"""### 指示
|
151 |
{input}
|
152 |
-
|
153 |
"""
|
154 |
|
155 |
tokenized_input = tokenizer.encode(prompt, add_special_tokens=False, return_tensors="pt").to("cuda")
|
|
|
149 |
|
150 |
prompt = f"""### 指示
|
151 |
{input}
|
152 |
+
### 回答
|
153 |
"""
|
154 |
|
155 |
tokenized_input = tokenizer.encode(prompt, add_special_tokens=False, return_tensors="pt").to("cuda")
|