id should be id

AI-Commandos · Apr 19, 2024 · a33e8b0 · a33e8b0
1 parent 457e0be
commit a33e8b0
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/finetune.py b/finetune.py
@@ -140,7 +140,7 @@ def main():
     if padding == 'left':
         tokenizer.pad_token_id = 0
     else:
-        tokenizer.pad_token_id = tokenizer.eos_token
+        tokenizer.pad_token_id = tokenizer.eos_token_id
     tokenizer.padding_side = padding
 
     # Set up LoRA configuration

diff --git a/finetune_dpo.py b/finetune_dpo.py
@@ -144,7 +144,7 @@ def main():
     if padding == 'left':
         tokenizer.pad_token_id = 0
     else:
-        tokenizer.pad_token_id = tokenizer.eos_token
+        tokenizer.pad_token_id = tokenizer.eos_token_id
     tokenizer.padding_side = padding
 
     # Set up LoRA configuration

diff --git a/finetune_orpo.py b/finetune_orpo.py
@@ -142,7 +142,7 @@ def main():
     if padding == 'left':
         tokenizer.pad_token_id = 0
     else:
-        tokenizer.pad_token_id = tokenizer.eos_token
+        tokenizer.pad_token_id = tokenizer.eos_token_id
     tokenizer.padding_side = padding
 
     orpo_config = ORPOConfig(