본문으로 건너뛰기
IFAI

오픈소스

Replit, 30억 파라미터 코드 생성 모델 'Replit Code V1.5 3B' 오픈소스 공개

게시 확산 지수 46

Replit이 허깅페이스에 코드 보완 모델 Replit Code V1.5 3B를 오픈소스로 공개했다. 1조 개 토큰과 30개 주요 언어를 학습했으며, 파인튜닝 시 CodeLlama 7B 등 더 큰 모델을 앞서는 성능을 낸다. 상업적 목적의 미세조정도 자유롭게 허용된다.

Replit, 30억 파라미터 코드 생성 모델 'Replit Code V1.5 3B' 오픈소스 공개
이미지: blog.replit.com

Replit Code V1.5 3B는 Replit이 오픈소스로 공개한 경량 코드 생성 모델입니다. 30억 파라미터(3B) 규모로 코드 완성(code completion) 작업에 특화되어 있으며, 상업적 목적의 미세조정(fine-tuning)도 별다른 제한 없이 무료로 허용됩니다.

외부 클라우드 API에 의존하지 않고 로컬 기기나 사내 서버에서 빠른 반응 속도의 코드 자동완성을 구축하려는 개발자에게 유용한 베이스 모델입니다.

기능과 성능

이 모델은 The Stack 데이터셋의 허용 라이선스 코드와 StackExchange의 공개 기술 문답을 포함한 1조 개의 토큰으로 사전 학습되었습니다.

학습 데이터셋은 구문 분석 가능 여부 검사, 코드 품질 휴리스틱 필터링, 유해 표현 제거 등 엄격한 정제 과정을 거쳤습니다.

지원 언어는 Replit 내 상위 30개 프로그래밍 언어를 망라하며, 최적화된 토큰 처리를 위해 32K 크기의 전용 어휘 사전을 탑재했습니다.

평가 결과, 3B 급 모델 가운데 HumanEval과 Multi-PLe 벤치마크에서 선두권 점수를 기록했습니다. Replit 측은 공개 코드 데이터로 미세조정할 경우 파라미터가 2배 이상 큰 CodeLlama 7B보다 뛰어난 성능을 보인다고 밝혔습니다.

기술적 특징과 실행 방법

낮은 추론 지연 시간과 높은 생성 품질을 달성하기 위해 Grouped Query Attention(GQA), Flash Attention Triton 커널, ALiBi 위치 임베딩 구조를 결합했습니다.

훈련 과정에는 LionW 옵티마이저, 학습률 쿨다운, QKV 클리핑 등 최신 안정화 기법이 두루 적용되었습니다.

모델은 허깅페이스(replit/replit-code-v1_5-3b)에서 바로 내려받을 수 있습니다. transformers 라이브러리의 AutoModelForCausalLM과 AutoTokenizer를 사용하면 몇 줄의 파이썬 코드로 손쉽게 구동 가능합니다.

GPU 연산 속도를 극대화할 수 있는 Triton 커널 최적화 설정법도 공식 저장소 문서에 함께 안내되어 있습니다.

왜 중요한가

Replit Code V1.5 3B는 소형 코드 모델의 실용성을 한 단계 끌어올렸다는 점에서 주목할 만합니다.

코드 자동완성 도구는 지연 시간이 조금만 길어져도 개발 흐름을 끊기 때문에 속도가 매우 중요합니다. 7B 이상의 모델은 로컬이나 소형 인스턴스에서 부담스럽지만, 3B 모델은 적은 VRAM으로도 가볍게 돌릴 수 있습니다.

대화형 챗봇 형태의 복잡한 소프트웨어 설계 용도는 아니지만, 사내 코드베이스나 특정 언어에 맞춰 미세조정한 뒤 가볍고 빠른 사내 자동완성 플러그인으로 활용하기에 적합합니다.

출처 (1)

  1. [1]
    blog.replit.com(새 탭에서 열림)원문blog.replit.com

매일 AI 콘텐츠 픽