update
Browse files
app.py
CHANGED
@@ -31,9 +31,16 @@ terminators = [
|
|
31 |
llama_tokenizer.eos_token_id,
|
32 |
llama_tokenizer.convert_tokens_to_ids("")
|
33 |
]
|
34 |
-
|
35 |
-
|
36 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
37 |
|
38 |
|
39 |
# Place just input pass and return generation output
|
|
|
31 |
llama_tokenizer.eos_token_id,
|
32 |
llama_tokenizer.convert_tokens_to_ids("")
|
33 |
]
|
34 |
+
|
35 |
+
# Get special tokens list from the tokenizer
|
36 |
+
special_tokens = llama_tokenizer.special_tokens_map
|
37 |
+
eos_token = special_tokens.get("eos_token")
|
38 |
+
|
39 |
+
# if the tokenizer doesn't find a specific EOS token, use the default one
|
40 |
+
if eos_token is None:
|
41 |
+
eos_token = llama_tokenizer.eos_token
|
42 |
+
|
43 |
+
print("EOS Token", eos_token)
|
44 |
|
45 |
|
46 |
# Place just input pass and return generation output
|