KICCO_AI_IMAGE/requirements.txt

48 lines
1.0 KiB
Plaintext

# ComfyUI 核心依赖
comfyui-frontend-package==1.16.9
comfyui-workflow-templates==0.1.3
torch==2.2.0 # 指定CUDA 12.1版本兼容
torchvision==0.17.0 # 指定CUDA 12.1版本兼容
torchaudio==2.2.0 # 指定CUDA 12.1版本兼容
torchsde==0.2.6
numpy==1.26.0 # flux版本是1.24.3,但我们使用更新版本
einops
transformers==4.51.3 # 用于CLIP模型和文本处理
tokenizers>=0.13.3
sentencepiece
safetensors>=0.4.2
yarl>=1.18.0
pyyaml
Pillow>=9.3.0 # 使用flux版本的最低要求
scipy
tqdm>=4.65.0 # 使用flux版本的最低要求
psutil
# FLUX风格塑形API依赖
fastapi>=0.102.0
uvicorn>=0.23.0
python-multipart>=0.0.6
huggingface_hub>=0.19.0
GitPython>=3.1.30
python-dotenv>=1.0.0
numba
colour-science
rembg
pixeloe
transparent-background
# MCP服务器依赖
mcp>=1.0.0 # Model Context Protocol Python SDK
# 非必要依赖
kornia>=0.7.1
spandrel
soundfile
av>=14.1.0
segment_anything
timm
addict
yapf
xformers==0.0.24
diffusers==0.33.1 # 用于Stable Diffusion和ControlNet模型
aiohttp>=3.11.13