Update README.md
Browse files
README.md
CHANGED
@@ -34,7 +34,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
|
|
34 |
model = model.half()
|
35 |
|
36 |
# Define special tokens
|
37 |
-
END_TOKEN = 64003 # -> [
|
38 |
B_INST, E_INST = "[INST]", "[/INST]"
|
39 |
SYSTEM_PROMPT = "A conversation between a human and an AI assistant."
|
40 |
|
|
|
34 |
model = model.half()
|
35 |
|
36 |
# Define special tokens
|
37 |
+
END_TOKEN = 64003 # -> [END]
|
38 |
B_INST, E_INST = "[INST]", "[/INST]"
|
39 |
SYSTEM_PROMPT = "A conversation between a human and an AI assistant."
|
40 |
|