Commit Graph

8 Commits

Author SHA1 Message Date
kooshi 8e471516b8
distributed data parallelism with torchrun 2023-03-24 23:56:06 -05:00
kooshi 2bc64597aa
model parallelism 2023-03-24 23:03:43 -05:00
John Smith 0879580006
Merge branch 'main' into finetune-refactor 2023-03-25 10:29:02 +08:00
Andrey Glushenkov 397f5041c3
Reflect last changes in main
Reflect commits:
4906961bf1
60b227d0ba
2023-03-24 15:46:03 +03:00
Andrey Glushenkov 50dbb101e9
Refactor finetune.py
1. Add command line arguments support
2. Add Stanford Alpaca-like dataset support. Used code from - https://github.com/tloen/alpaca-lora
3. Fix LoRA pre-train application
2023-03-24 14:15:07 +03:00
John Smith 60b227d0ba fix minor bug 2023-03-23 08:43:18 +00:00
John Smith 44978669cf Add gradient checkpointing 2023-03-23 08:25:29 +00:00
John Smith dc036373b2 add more scripts and adjust code for transformer branch 2023-03-22 04:09:04 +00:00