Spaces:
Runtime error
Runtime error
zetavg
commited on
Commit
•
7b14813
1
Parent(s):
c15d0e4
this should be set for training
Browse files- 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()
|