Yhhxhfh commited on
Commit
6838103
·
verified ·
1 Parent(s): 020fd51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -19,7 +19,9 @@ load_dotenv()
19
  HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
20
 
21
  global_data = {
22
-
 
 
23
  'model_configs': [
24
  {"repo_id": "Ffftdtd5dtft/gpt2-xl-Q2_K-GGUF", "filename": "gpt2-xl-q2_k.gguf", "name": "GPT-2 XL"},
25
  {"repo_id": "Ffftdtd5dtft/gemma-2-27b-Q2_K-GGUF", "filename": "gemma-2-27b-q2_k.gguf", "name": "Gemma 2-27B"},
 
19
  HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
20
 
21
  global_data = {
22
+ 'tokens': {'eos': 'eos_token', 'pad': 'pad_token', 'padding': 'padding_token',
23
+ 'unk': 'unk_token', 'bos': 'bos_token', 'sep': 'sep_token',
24
+ 'cls': 'cls_token', 'mask': 'mask_token'},
25
  'model_configs': [
26
  {"repo_id": "Ffftdtd5dtft/gpt2-xl-Q2_K-GGUF", "filename": "gpt2-xl-q2_k.gguf", "name": "GPT-2 XL"},
27
  {"repo_id": "Ffftdtd5dtft/gemma-2-27b-Q2_K-GGUF", "filename": "gemma-2-27b-q2_k.gguf", "name": "Gemma 2-27B"},