Kaynağa Gözat

add new argument: tokenizer_name

Rahul A R 11 ay önce
ebeveyn
işleme
2fa8e69b62
1 değiştirilmiş dosya ile 1 ekleme ve 0 silme
  1. 1 0
      src/llama_recipes/configs/training.py

+ 1 - 0
src/llama_recipes/configs/training.py

@@ -7,6 +7,7 @@ from dataclasses import dataclass
 @dataclass
 class train_config:
     model_name: str="PATH/to/LLAMA/7B"
+    tokenizer_name: str=None
     enable_fsdp: bool=False
     low_cpu_fsdp: bool=False
     run_validation: bool=True