Add 01-ai/Yi-6B (#2037)
Summary:
Add the model to canary as its CPU test exceeds time limit and GPU test will OOM on A100.
Pull Request resolved: https://github.com/pytorch/benchmark/pull/2037
Reviewed By: aaronenyeshi
Differential Revision: D51356557
Pulled By: xuzhao9
fbshipit-source-id: f462c27b787bb81fae3cf0631a001799ae462f6f