alfredplpl commited on
Commit
c59f6b0
1 Parent(s): a58acfb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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