Go to file
2024-10-28 11:02:38 +00:00
src process group order 2024-10-27 04:56:54 +00:00
.gitignore tesnsor parallel, will clean later 2024-10-18 05:13:44 +00:00
convert_hf_to_picotron.py various fix (modeling, dataloader, cpu load) 2024-10-18 14:33:46 +00:00
convert_picotron_to_hf.py refactor organisation 2024-10-10 15:12:14 +00:00
generate.py various fix (modeling, dataloader, cpu load) 2024-10-18 14:33:46 +00:00
model.py add cuda kernels 2024-10-22 22:38:29 +00:00
README.md Initial commit 2024-09-18 14:01:22 +02:00
requirements.txt add wandb support 2024-09-25 14:19:16 +00:00
setup.py tesnsor parallel, will clean later 2024-10-18 05:13:44 +00:00
train.py typo 2024-10-28 11:02:38 +00:00
utils.py save/load weights 2024-10-28 05:19:59 +00:00

picotron