my-image-generation / requirements.txt
i0switch's picture
Update requirements.txt
f96149f verified
raw
history blame
918 Bytes
--extra-index-url https://download.pytorch.org/whl/cu118 # GPU wheel を取得
# ==== ZeroGPU & Gradio ====
gradio[mcp,oauth]==5.34.2
spaces==0.37.1
huggingface-hub>=0.30,<0.40 # gradio が要求
# ==== ML Core ====
torch==2.4.0 # ★ ZeroGPU 対応世代 :contentReference[oaicite:2]{index=2}
torchvision==0.19.1 # torch-2.4 用ビルド :contentReference[oaicite:3]{index=3}
xformers==0.0.28.post1 # torch-2.4 向け wheel :contentReference[oaicite:4]{index=4}
diffusers==0.30.3 # cached_download 不要バージョン
accelerate==0.29.3
transformers==4.41.1
peft==0.11.1
safetensors>=0.5.3
numpy==1.26.4 # NumPy-2 ABI 衝突回避
# ==== CV / 画像処理 ====
opencv-python-headless
Pillow
insightface
basicsr
realesrgan
onnxruntime-gpu
# ==== API / ASGI ====
fastapi
python-multipart