Procházet zdrojové kódy

add new argument: tokenizer_name

Rahul A R před 11 měsíci
rodič
revize
2fa8e69b62
1 změnil soubory, kde provedl 1 přidání a 0 odebrání
  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