alfredplpl
commited on
Commit
•
c59f6b0
1
Parent(s):
a58acfb
Update app.py
Browse files
app.py
CHANGED
@@ -49,6 +49,7 @@ h1 {
|
|
49 |
tokenizer = AutoTokenizer.from_pretrained("tokyotech-llm/Llama-3-Swallow-8B-Instruct-v0.1")
|
50 |
model = AutoModelForCausalLM.from_pretrained("tokyotech-llm/Llama-3-Swallow-8B-Instruct-v0.1", torch_dtype=torch.bfloat16, device_map="auto")
|
51 |
terminators = [
|
|
|
52 |
tokenizer.convert_tokens_to_ids("<|eot_id|>")
|
53 |
]
|
54 |
|
|
|
49 |
tokenizer = AutoTokenizer.from_pretrained("tokyotech-llm/Llama-3-Swallow-8B-Instruct-v0.1")
|
50 |
model = AutoModelForCausalLM.from_pretrained("tokyotech-llm/Llama-3-Swallow-8B-Instruct-v0.1", torch_dtype=torch.bfloat16, device_map="auto")
|
51 |
terminators = [
|
52 |
+
tokenizer.eos_token_id,
|
53 |
tokenizer.convert_tokens_to_ids("<|eot_id|>")
|
54 |
]
|
55 |
|