CreitinGameplays commited on
Commit
c76b802
·
verified ·
1 Parent(s): d6af013

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -32,7 +32,8 @@ def load_model():
32
  torch_dtype=torch.float16,
33
  device_map="auto"
34
  )
35
-
 
36
  tokenizer = AutoTokenizer.from_pretrained(model_id, padding_side='left')
37
  tokenizer.use_default_system_prompt = False
38
 
 
32
  torch_dtype=torch.float16,
33
  device_map="auto"
34
  )
35
+ tokenizer = Tokenizer(BPE())
36
+ tokenizer.pre_tokenizer = Whitespace()
37
  tokenizer = AutoTokenizer.from_pretrained(model_id, padding_side='left')
38
  tokenizer.use_default_system_prompt = False
39