KICCO_AI_IMAGE/requirements.txt

48 lines
1.0 KiB
Plaintext
Raw Permalink Normal View History

2025-05-02 20:45:07 +08:00
# ComfyUI 核心依赖
comfyui-frontend-package==1.16.9
2025-04-20 03:21:46 +08:00
comfyui-workflow-templates==0.1.3
2025-05-02 20:54:24 +08:00
torch==2.2.0 # 指定CUDA 12.1版本兼容
torchvision==0.17.0 # 指定CUDA 12.1版本兼容
torchaudio==2.2.0 # 指定CUDA 12.1版本兼容
2025-04-25 11:18:48 +08:00
torchsde==0.2.6
2025-05-02 20:45:07 +08:00
numpy==1.26.0 # flux版本是1.24.3,但我们使用更新版本
2023-01-03 14:53:32 +08:00
einops
2025-05-02 20:45:07 +08:00
transformers==4.51.3 # 用于CLIP模型和文本处理
2024-07-12 04:51:06 +08:00
tokenizers>=0.13.3
2024-07-12 10:58:03 +08:00
sentencepiece
safetensors>=0.4.2
yarl>=1.18.0
pyyaml
2025-05-02 20:45:07 +08:00
Pillow>=9.3.0 # 使用flux版本的最低要求
scipy
2025-05-02 20:45:07 +08:00
tqdm>=4.65.0 # 使用flux版本的最低要求
psutil
2024-05-27 01:44:47 +08:00
2025-05-02 20:45:07 +08:00
# FLUX风格塑形API依赖
2025-04-24 19:25:29 +08:00
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
2025-05-02 20:45:07 +08:00
# MCP服务器依赖
2025-04-24 19:25:29 +08:00
mcp>=1.0.0 # Model Context Protocol Python SDK
2025-05-02 20:45:07 +08:00
# 非必要依赖
kornia>=0.7.1
2024-05-27 01:44:47 +08:00
spandrel
soundfile
av>=14.1.0
2025-05-02 20:43:42 +08:00
segment_anything
timm
addict
yapf
xformers==0.0.24
2025-05-02 20:45:07 +08:00
diffusers==0.33.1 # 用于Stable Diffusion和ControlNet模型
2025-05-03 11:46:43 +08:00
aiohttp>=3.11.13