DeepSpeed
Add unit test for various model families and inference tasks
#1981
Merged

Add unit test for various model families and inference tasks #1981

mrwyattii
mrwyattii added unit test for various HF model families and tasks
ddf4600d
mrwyattii formatting
e95b44d8
mrwyattii added missing import
9fca22cb
mrwyattii fixed broken pytest global vars
32d3e49b
mrwyattii modified test to conform to other test structure
aee4b5a7
jeffra Merge branch 'master' into add-hf-model-tests
42a2569e
mrwyattii removed gpt-j. it cannot run on V100s (OOM)
ddb97025
jeffra jeffra marked this pull request as ready for review 4 years ago
jeffra jeffra requested a review from jeffra jeffra 4 years ago
jeffra jeffra requested a review from samyam samyam 4 years ago
jeffra jeffra requested a review from tjruwase tjruwase 4 years ago
jeffra jeffra requested a review from ShadenSmith ShadenSmith 4 years ago
jeffra jeffra requested a review from conglongli conglongli 4 years ago
jeffra jeffra requested a review from awan-10 awan-10 4 years ago
jeffra jeffra requested a review from cli99 cli99 4 years ago
jeffra jeffra requested a review from eltonzheng eltonzheng 4 years ago
jeffra jeffra requested a review from minjiaz minjiaz 4 years ago
jeffra jeffra requested a review from RezaYazdaniAminabadi RezaYazdaniAminabadi 4 years ago
jeffra
jeffra approved these changes on 2022-05-31
mrwyattii mrwyattii merged 5d3c6718 into master 4 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone