zetavg commited on
Commit
7b14813
1 Parent(s): c15d0e4

this should be set for training

Browse files
Files changed (1) hide show
  1. llama_lora/models.py +4 -0
llama_lora/models.py CHANGED
@@ -101,6 +101,10 @@ def load_base_model():
101
  Global.base_model, device_map={"": device}, low_cpu_mem_usage=True
102
  )
103
 
 
 
 
 
104
 
105
  def clear_cache():
106
  gc.collect()
 
101
  Global.base_model, device_map={"": device}, low_cpu_mem_usage=True
102
  )
103
 
104
+ Global.base_model.config.pad_token_id = get_tokenizer().pad_token_id = 0
105
+ Global.base_model.config.bos_token_id = 1
106
+ Global.base_model.config.eos_token_id = 2
107
+
108
 
109
  def clear_cache():
110
  gc.collect()