erikbeltran commited on
Commit
e9c8f39
·
verified ·
1 Parent(s): c8ab0ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -15,13 +15,13 @@ try:
15
  tokenizer = PreTrainedTokenizerFast.from_pretrained(MODEL_ID)
16
 
17
  # Ensure the tokenizer has the necessary special tokens
18
- special_tokens = {
19
- 'pad_token': '[PAD]',
20
- 'eos_token': '</s>',
21
- 'bos_token': '<s>',
22
- 'unk_token': '<unk>'
23
- }
24
- tokenizer.add_special_tokens(special_tokens)
25
 
26
  model = AutoModelForCausalLM.from_pretrained(MODEL_ID, gguf_file=GGUF_FILE)
27
 
 
15
  tokenizer = PreTrainedTokenizerFast.from_pretrained(MODEL_ID)
16
 
17
  # Ensure the tokenizer has the necessary special tokens
18
+ #special_tokens = {
19
+ # 'pad_token': '[PAD]',
20
+ # 'eos_token': '</s>',
21
+ # 'bos_token': '<s>',
22
+ # 3 'unk_token': '<unk>'
23
+ #}
24
+ #tokenizer.add_special_tokens(special_tokens)
25
 
26
  model = AutoModelForCausalLM.from_pretrained(MODEL_ID, gguf_file=GGUF_FILE)
27