Google Colab에서 Hugging Face 모델 사용하기

Hugging Face 모델을 Google Colab에서 사용하는 방법

Google Colab은 무료로 Hugging Face 모델을 실행할 수 있는 편리한 플랫폼입니다. 이 가이드는 Google Colab을 사용하여 Hugging Face의 "Illusion Diffusion" 모델을 설정하고 실행하는 방법을 단계별로 안내합니다.

1. Google Colab 준비하기

1.1 Colab 노트북 생성

1. Google Colab 페이지를 열고 Google 계정으로 로그인합니다.
2. 파일(File) > 새 노트북(New Notebook)을 클릭하여 새 노트북을 만듭니다.
3. 노트북 이름을 지정합니다.

1.2 런타임 설정 변경

1. 런타임(Runtime) > 런타임 유형 변경(Change Runtime Type)을 클릭합니다.
2. 하드웨어 가속기(Hardware Accelerator)를 GPU로 변경하고 저장합니다.

2. Hugging Face 모델 복제하기

2.1 모델 저장소 복제

1. Hugging Face 웹사이트에서 사용할 모델의 저장소 링크를 복사합니다. 예시: Illusion Diffusion.
2. Colab 노트북에서 아래 명령어를 실행하여 모델 저장소를 복제합니다:

!git clone https://huggingface.co/{모델_저장소_주소}

2.2 작업 디렉토리 이동

복제한 모델 디렉토리로 이동합니다:

%cd {모델_폴더_이름}

2.3 디렉토리 파일 확인

저장소의 파일을 확인하려면 다음 명령어를 사용합니다:

!ls

3. 필수 패키지 설치

3.1 requirements.txt 파일 설치

모델에 필요한 Python 패키지는 requirements.txt에 나열되어 있습니다. 이를 설치하려면 다음 명령어를 실행하세요:

!pip install -r requirements.txt -q

아래는 requirements.txt 파일의 예제 내용입니다:

diffusers
transformers
accelerate
xformers
Pillow
qrcode
filelock
--extra-index-url https://download.pytorch.org/whl/cu118
torch

3.2 추가 패키지 설치

일부 필수 패키지가 requirements.txt에 누락된 경우 수동으로 설치해야 합니다. 예를 들어, Gradio가 필요하다면 다음 명령어를 실행합니다:

!pip install gradio

4. 모델 실행 준비

4.1 Hugging Face 토큰 생성

1. Hugging Face 웹사이트에서 계정에 로그인합니다.
2. 설정(Settings) > Access Tokens로 이동합니다.
3. 새 토큰 생성(Create New Token)을 클릭하고 "Read" 권한을 선택합니다.
4. 생성된 토큰을 복사합니다.

4.2 토큰 설정

Colab에서 다음 코드를 실행하여 Hugging Face에 로그인합니다:

from huggingface_hub import notebook_login
notebook_login()

프롬프트가 나타나면 토큰을 붙여넣고 로그인합니다.

5. app.py 파일 실행

5.1 코드 수정

app.py 파일을 열고 아래 내용을 추가합니다:

app.launch(share=True)

이 코드는 Gradio 애플리케이션을 실행하고 외부에서 접근할 수 있는 링크를 제공합니다.

5.2 파일 실행

아래 명령어로 app.py를 실행합니다:

!python app.py

6. 에러 해결

6.1 누락된 패키지 설치

ModuleNotFoundError가 발생하면 누락된 패키지를 수동으로 설치해야 합니다. 예를 들어, "spaces" 패키지가 누락된 경우:

!pip install spaces

6.2 런타임 오류

Hugging Face 토큰 인증과 관련된 오류가 발생하면 토큰을 확인하고 다시 입력하세요.

7. 모델 테스트

7.1 링크 접근

app.py 실행 후 생성된 링크(공용 URL)를 통해 Gradio 인터페이스에 접근할 수 있습니다. 공용 URL은 72시간 동안 유효합니다.

7.2 테스트 입력

1. 인터페이스에서 이미지를 업로드합니다.
2. 프롬프트를 입력합니다(예: "snow forest").
3. 실행 결과를 확인합니다.