-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathrequirements.txt
More file actions
141 lines (108 loc) · 3.65 KB
/
Copy pathrequirements.txt
File metadata and controls
141 lines (108 loc) · 3.65 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
# Photo666 v0.3.0 Dependencies
# ========================================
# 核心依赖
# ========================================
# Core GUI Framework
PyQt6>=6.5.0
# Image Processing
Pillow>=10.2.0
pillow>=10.2.0 # 兼容性别名
# Configuration Management
PyYAML>=6.0.0
# Database Utilities
sqlite-utils>=3.35.0
# Translation Support
googletrans-py>=4.0.0
# Async Support
asyncio-mqtt>=0.11.0
aiofiles>=22.0.0
# ========================================
# AI模型依赖
# ========================================
# PyTorch (CUDA 12.8兼容版本，支持GPU推理)
# 注意：这些版本专门为CUDA 12.8优化，确保GPU加速功能
# 版本已根据实际虚拟环境配置调整
torch>=2.7.0+cu128
torchvision>=0.22.0+cu128
torchaudio>=2.7.0+cu128
# 如果上述版本安装失败，可以使用以下替代方案：
# 方案1：从PyTorch官方安装（推荐）
# pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
# 方案2：使用conda安装
# conda install pytorch torchvision torchaudio pytorch-cuda=12.8 -c pytorch -c nvidia
# Transformers (AI模型框架)
# 版本已根据实际虚拟环境配置调整
transformers>=4.55.0
# Hugging Face Hub
huggingface_hub>=0.16.0
# Accelerate (模型加速)
# 版本已根据实际虚拟环境配置调整
accelerate>=1.9.0
# Safetensors (模型文件格式)
safetensors>=0.3.0
# Bitsandbytes (量化支持)
bitsandbytes>=0.41.0
# ========================================
# 计算机视觉依赖
# ========================================
# NumPy (数值计算)
numpy>=1.24.0
# Einops (张量操作)
einops>=0.8.0
# Timm (图像模型)
timm>=1.0.0
# ========================================
# 网络和工具依赖
# ========================================
# HTTP请求
requests>=2.28.0
# 地理计算
geopy>=2.3.0
# 进度条
tqdm>=4.60.0
# ========================================
# 开发依赖 (可选)
# ========================================
# Testing
pytest>=7.0.0
pytest-cov>=4.0.0
pytest-qt>=4.2.0
# Code Quality
black>=23.0.0
mypy>=1.0.0
# Documentation
sphinx>=5.0.0
sphinx-rtd-theme>=1.2.0
# Packaging
PyInstaller>=5.0.0
setuptools>=65.0.0
wheel>=0.38.0
# Testing Utilities
factory-boy>=3.2.0
faker>=18.0.0
# ========================================
# 插件依赖整合
# ========================================
# 以下依赖涵盖了所有插件的需求：
# - florence2_reverse_plugin: transformers, torch, torchvision, accelerate, safetensors, pillow, numpy, einops, timm
# - joycaption_reverse_plugin: torch, transformers, huggingface_hub, pillow, requests, bitsandbytes, accelerate, safetensors
# - janus_reverse_plugin: janus, torch, transformers, pillow, numpy, requests, tqdm
# - janus_text2image_plugin: pillow
# - gps_location_plugin: requests, geopy
# ========================================
# 特殊依赖
# ========================================
# Janus AI模型代码已集成在janus_text2image_plugin插件中
# 无需从GitHub重复安装
# ========================================
# 注意事项
# ========================================
# 1. 首次安装可能需要较长时间，因为需要下载AI模型
# 2. 建议在虚拟环境中安装
# 3. 如果遇到网络问题，可以使用国内镜像源
# 4. CUDA版本要求：需要CUDA 12.8或兼容版本
# 5. GPU支持：确保安装CUDA版本的PyTorch以获得GPU加速
# 6. 如果pip安装CUDA版本失败，请使用PyTorch官方命令：
# pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
# 7. Janus相关功能已集成在插件中，无需额外安装
# 8. 验证GPU支持：python -c "import torch; print('CUDA available:', torch.cuda.is_available())"