requirements.txt

# GPT2-BERT 병합 모델 테스터 Requirements

# 필수 라이브러리
torch>=2.0.0
transformers>=4.35.0
gradio
numpy>=1.24.0

# GPU 지원 (선택사항)
# CUDA 11.8 기준 - 본인의 CUDA 버전에 맞게 수정
# torch==2.0.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118

# 추가 유틸리티
huggingface-hub>=0.19.0
accelerate>=0.25.0
sentencepiece>=0.1.99
protobuf>=3.20.0

# 모델 최적화 (선택사항)
bitsandbytes>=0.41.0