Matthias Reso 33925f71e6 Add missing amp context if use_fp16 is enabled 2 years ago
..
configs cc356b6017 Add FSDP CPU offloading option 3 years ago
datasets b8d327f7c3 Update grammar_dataset_process.ipynb 2 years ago
inference e554c1c8bf The tokenizer will not add eos_token by default 3 years ago
model_checkpointing ce9501f22c remove relative imports 3 years ago
policies ce9501f22c remove relative imports 3 years ago
utils 33925f71e6 Add missing amp context if use_fp16 is enabled 2 years ago
finetuning.py 91e2573aa8 pass weight_decay into optimizer 3 years ago