[HuggingFace] 모델 업로드하기

허깅페이스(Hugging Face)는 모델과 데이터셋을 공유하고 관리하는 데 사용되는 플랫폼입니다. 모델을 허깅페이스에 업로드하려면 다음 단계를 따르면 됩니다.

1. 허깅페이스 CLI 설치

허깅페이스 CLI를 설치해야 합니다. 터미널 또는 명령 프롬프트에서 다음 명령어를 실행하여 설치할 수 있습니다.

pip install huggingface-cli

2. 로그인

허깅페이스 CLI를 사용하여 계정에 로그인합니다.

huggingface-cli login

3. 새로운 레포지토리 생성 (옵션)

모델을 업로드할 새로운 레포지토리를 만들 수 있습니다. 이미 존재하는 레포지토리에 업로드할 수도 있습니다.

huggingface-cli repo create <repository_name>

4. 모델 업로드

모델 파일을 허깅페이스에 업로드합니다. 이때 <repository_name>은 업로드할 레포지토리의 이름이고, <model_directory>는 모델 파일이 저장된 디렉토리입니다.

cd <model_directory>
huggingface-cli upload <repository_name> <model_directory>

위 명령어에서 checkpoint-best/는 모델 파일이 저장된 디렉토리를 나타냅니다. 이 디렉토리에는 모델 파일 외에도 모델에 필요한 다른 파일들이 포함되어 있어야 합니다.

5. 업로드 확인

업로드가 성공적으로 완료되면 해당 레포지토리의 페이지에서 모델을 확인할 수 있습니다.

위 단계를 따르면 모델을 허깅페이스에 업로드할 수 있습니다. 만약 새로운 레포지토리를 생성하지 않고 기존의 레포지토리에 업로드하려면 단계 3을 건너뛰고 단계 4에서 레포지토리 이름 대신 기존 레포지토리의 이름을 사용하면 됩니다.