피드로 돌아가기
Hugging Face + PyCharm
Hugging Face BlogHugging Face Blog
AI/ML

JetBrains가 PyCharm에 Hugging Face 통합 기능을 추가해 모델 검색·삽입·캐싱을 IDE 내에서 직접 수행 가능하게 함

Hugging Face + PyCharm

2024년 11월 5일6beginner

Context

파이썬 개발자가 머신러닝 모델을 코드에 적용할 때 Hugging Face Hub 웹사이트를 방문해야 하고, 모델 카드를 검색한 후 샘플 코드를 복사해 붙여넣는 과정이 필요했다. 코드 리뷰 시에도 사용된 모델의 출처와 의도를 확인하기 위해 외부 문서를 참고해야 했다.

Technical Solution

  • 우클릭 메뉴에서 "Insert HF Model" 옵션을 통해 IDE 내 대화창에서 모델 검색: 모델명, 최종 업데이트 날짜, Like 수 기준 정렬 기능 제공
  • 선택한 모델의 전체 Model Card를 대화창 우측에 표시하고 샘플 코드를 직접 복사해 에디터에 붙여넣기 기능 구현
  • 모델명 위에 마우스를 호버할 때 Model Card 팝업을 즉시 표시해 코드 리뷰 시 모델 검증 가능하도록 구현
  • 로컬 모델 캐시 관리 UI 추가: 🤗 아이콘을 클릭하면 다운로드된 모델 목록 확인 및 디스크 공간 확보 목적으로 캐시 삭제 가능

Key Takeaway

IDE 통합을 통해 오픈소스 AI 모델을 일반 라이브러리 임포트만큼 접근 가능하게 함으로써, 머신러닝 추상화 계층을 프로그래밍 워크플로우의 자연스러운 부분으로 만드는 것이 향후 파이썬 개발 환경의 방향성이다.


PyCharm Professional을 사용하는 파이썬 엔지니어가 이미지-텍스트 기반 챗봇, 이미지 인식 등 다양한 머신러닝 기능을 IDE를 벗어나지 않고 10분 이내에 작동 가능한 코드로 통합할 수 있으므로, 프로토타이핑 속도를 크게 단축할 수 있다.

원문 읽기