Tri Dao 9d797d8848 Support loading GPT2 weights from Huggingface		2022-12-27 11:22:48 -08:00
..
test_bert.py	Tweak CrossEntropyLoss to take process_group in init	2022-12-27 10:47:43 -08:00
test_gpt_parallel.py	Implement Tensor Parallel for GPT model	2022-12-26 16:22:43 -08:00
test_gpt.py	Support loading GPT2 weights from Huggingface	2022-12-27 11:22:48 -08:00