Commit Graph

73 Commits (c46e09715cc906b9c82a92f1287bd1b1defe87ea)

Author SHA1 Message Date
Baizhou Zhang 2a2eacfaf1 [pipeline] support shardformer for GPT2ForQuestionAnswering & complete pipeline support for GPT2 (#4245)
1 year ago
Jianghai e7cc62d735 [pipeline] All bert models (#4233)
1 year ago
Jianghai 37d22f6878 [pipeline] add bloom model pipeline (#4210)
1 year ago
Jianghai 31bcf867ae [pipeline] Llama causal lm and llama for sequence classification pipeline (#4208)
1 year ago
Jianghai 1622031058 [pipeline] Llama pipeline (#4205)
1 year ago
Frank Lee ae035d305d [shardformer] added embedding gradient check (#4124)
1 year ago
Frank Lee b1c2901530 [shardformer] supported bloom model (#4098)
1 year ago
jiangmingyan ac80937138 [shardformer] shardformer support opt models (#4091)
1 year ago
FoolPlayer 7740c55c55 support kit use for bert/gpt test (#4055)
1 year ago
Frank Lee 58df720570 [shardformer] adapted T5 and LLaMa test to use kit (#4049)
1 year ago
digger yu e61ffc77c6
fix typo tests/ (#3936)
1 year ago
Hongxin Liu 4b3240cb59
[booster] add low level zero plugin (#3594)
2 years ago
YuliangLiu0306 f57d34958b
[FX] refactor experimental tracer and adapt it with hf models (#3157)
2 years ago
Frank Lee 085e7f4eff
[test] fixed torchrec registration in model zoo (#3177)
2 years ago
Frank Lee 1ad3a636b1
[test] fixed torchrec model test (#3167)
2 years ago
ver217 6ae8ed0407
[lazyinit] add correctness verification (#3147)
2 years ago
Frank Lee ed19290560
[booster] implemented mixed precision class (#3151)
2 years ago
YuliangLiu0306 ecd643f1e4
[test] add torchrec models to test model zoo (#3139)
2 years ago
ver217 14a115000b
[tests] model zoo add torchaudio models (#3138)
2 years ago
Frank Lee 6d48eb0560
[test] added transformers models to test model zoo (#3135)
2 years ago
Frank Lee a674c63348
[test] added torchvision models to test model zoo (#3132)
2 years ago
HELSON 1216d1e7bd
[tests] diffuser models in model zoo (#3136)
2 years ago
Frank Lee 86ac782d7c
[test] added timm models to test model zoo (#3129)
2 years ago