Yi 시리즈 모델

Yi 시리즈 모델

Yi 는 01.AI 가 처음부터(scratch) 직접 훈련한 오픈소스 LLM 시리즈예요. 영어와 중국어 양쪽을 목표로 한 이중언어 모델이고, Apache 2.0 라이선스라 상업적으로도 자유롭게 쓸 수 있어요.

출처: https://github.com/01-ai/Yi

주요 모델

모델 크기 기본 컨텍스트
Yi-6B / Yi-6B-Chat 6B 4K (추론 시 32K 확장)
Yi-9B 9B — (Yi-6B 기반 지속학습, 코딩·수학 강점)
Yi-34B / Yi-34B-Chat 34B 4K
Yi-34B-200K / Yi-6B-200K 34B / 6B 200K

200K 컨텍스트는 대략 중국어 40만 자에 해당하는 길이예요.

성능 하이라이트

출시 당시 벤치마크 기준을 보면, Yi-34B-Chat 은 AlpacaEval 리더보드에서 GPT-4 Turbo 다음인 2위에 올랐고, Yi-34B 는 영어·중국어 오픈소스 모델 중 1위를 기록했어요(Falcon-180B, Llama-70B 등을 제치고).

어떻게 시작하나요

공식 저장소를 클론하고 요구사항을 설치한 뒤, 챗 모델은 AutoModelForCausalLM(transformers)로, 베이스 모델은 text_generation.py 로 실행할 수 있어요. 아키텍처가 Llama 와 호환돼서 기존 도구·라이브러리를 그대로 쓸 수 있는 게 큰 장점이에요.

더 알아보기