Kai Wu ee204ccb98 working now 9 months ago
..
__init__.py aa24e8f57e fix typos 1 year ago
config_utils.py b566582a86 finetune not working with fsdp 9 months ago
dataset_utils.py 1a183c0a5e Introduce Llama guard customization notebook and associated dataset loader example 10 months ago
flop_utils.py 26e877fd42 changed readme, unified the context interface and added get_flops_per_sec() 1 year ago
fsdp_utils.py ee204ccb98 working now 9 months ago
memory_utils.py c5a382e509 Make tests run on cpu only machines 1 year ago
plot_metrics.py 6d449a859b New folder structure (#1) 1 year ago
train_utils.py ee204ccb98 working now 9 months ago