허깅 페이스 PyCharm 통합
허깅 페이스와 JetBrains는 허깅 페이스 허브를 PyCharm Professional에 직접 통합하여, 개발자가 통합 개발 환경(IDE) 내에서 머신러닝 모델을 검색, 구현, 관리할 수 있도록 합니다. 이 통합은 웹 브라우저와 IDE 간의 전환 마찰을 줄이고, 오픈소스 모델을 소프트웨어 개발을 위한 접근 가능한 함수처럼 취급합니다.
통합 모델 발견 및 삽입
개발자는 이제 PyCharm 편집기에서 머신러닝 모델을 직접 발견하고 삽입할 수 있습니다. 코드에서 우클릭하고 "HF 모델 삽입"을 선택하면, 다음과 같은 기능을 제공하는 대화 상자가 나타납니다:
- 검색 및 필터: 카테고리(예: "image-text-to-text))별로 모델을 찾아보고, 인기(좋아요) 또는 최신성(마지막 업데이트 날짜)으로 정렬합니다.
- 모델 카드 접근: IDE 내에서 허깅 페이스 허브 모델 카드를 전체적으로 보아 문서화 및 예정된 용도를 검토합니다.
- 직접 코드 구현: 모델 카드에서 샘플 코드를 직접 복사하여 노트북 또는 스크립트에 붙여넣어 모델을 빠르게 구현합니다.
호버를 통한 즉시 모델 문서화
코드 리뷰 및 유지보수를 향상시키기 위해, 이 통합은 모델 문서에 즉시 접근할 수 있도록 제공합니다. 개발자가 코드 내에서 모델 이름에 호버하면 관련된 모델 카드가 자동으로 나타납니다. 이를 통해 동료나 리뷰어는 외부 브라우저에 접근하지 않고도 특정 모델의 출처와 예정된 용도를 확인할 수 있습니다.
로컬 모델 캐시 관리
PyCharm은 이제 IDE 내의 허깅 페이스 아이콘을 통해 접근 가능한 로컬 모델 캐시의 시각적 인터페이스를 포함합니다. 이 도구를 통해 개발자는 다음을 수행할 수 있습니다:
- 모델 인벤토리: 로컬 캐시에 현재 저장된 모든 모델의 전체 목록을 확인합니다.
- 디스크 공간 관리: 사용하지 않는 모델을 식별하고 삭제하여 디스크 공간을 확보합니다. 이는 프로덕션 준비 모델이 종종 1GB를 초과하기 때문에 중요합니다.
모델 실행 시 기술적 고려 사항
이 통합을 통해 모델을 구현할 때 개발자는 다음과 같은 하드웨어 제약 사항에 유의해야 합니다:
- 메모리 오류: 메모리 오류가 발생하면, 더 많은 메모리를 가진 GPU를 사용하거나 샘플 코드의 매개변수를 줄이는 것이 좋습니다.
- 하드웨어 오프로딩: 코드에서
device_map="cuda"를 제거하면 모델이 CPU 메모리에서 실행되도록 허용되지만, 이로 인해 실행 속도가 느려집니다.
Sources
- OriginalHugging Face + PyCharm