ソースを参照

add new argument: tokenizer_name

Rahul A R 11 ヶ月 前
コミット
2fa8e69b62
1 ファイル変更1 行追加0 行削除
  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