| .. |
|
hf_llama_conversion
|
6d449a859b
New folder structure (#1)
|
1 year ago |
|
__init__.py
|
aa24e8f57e
fix typos
|
1 year ago |
|
config_utils.py
|
17a6d16289
Test batching for both llama versions
|
1 year ago |
|
dataset_utils.py
|
fe8122daf1
Adapt alpaca dataset to ConcatDataset
|
2 years ago |
|
flop_utils.py
|
26e877fd42
changed readme, unified the context interface and added get_flops_per_sec()
|
1 year ago |
|
fsdp_utils.py
|
e9d5cfdfd5
Update src/llama_recipes/utils/fsdp_utils.py
|
1 year 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
|
26e877fd42
changed readme, unified the context interface and added get_flops_per_sec()
|
1 year ago |
Kai Wu
26e877fd42
changed readme, unified the context interface and added get_flops_per_sec()