-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathrequirements.txt
More file actions
62 lines (56 loc) · 1.96 KB
/
requirements.txt
File metadata and controls
62 lines (56 loc) · 1.96 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
# Core Dependencies
fastapi==0.104.1
uvicorn[standard]==0.24.0
websockets==12.0
slowapi==0.1.9
# =============================================================================
# PyTorch Installation - CHOOSE ONE OPTION BASED ON YOUR GPU
# =============================================================================
#
# OPTION 1: NVIDIA GPU (CUDA 12.1) - Default
# ------------------------------------------
# pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
#
# OPTION 2: NVIDIA GPU (CUDA 11.8) - Older GPUs
# ---------------------------------------------
# pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
#
# OPTION 3: AMD GPU (ROCm 6.0) - Linux only
# -----------------------------------------
# pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.0
#
# OPTION 4: Intel Arc GPU (XPU)
# -----------------------------
# pip install torch torchvision torchaudio intel-extension-for-pytorch --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
#
# OPTION 5: Apple Silicon (MPS) - macOS
# -------------------------------------
# pip install torch torchvision torchaudio
# (MPS support is built into the default PyTorch for macOS)
#
# OPTION 6: CPU Only (No GPU)
# ---------------------------
# pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
#
# =============================================================================
# Default: NVIDIA CUDA 12.1 (comment out if using different GPU)
--extra-index-url https://download.pytorch.org/whl/cu121
torch>=2.0.0
torchvision>=0.15.0
torchaudio>=2.0.0
# Diffusion & AI Libraries
diffusers==0.30.3
transformers==4.40.0
accelerate==0.28.0
huggingface_hub>=0.24.0
# Additional dependencies
Pillow==10.1.0
numpy==1.24.3
loguru==0.7.2
python-dotenv==1.0.0
psutil>=5.9.0
# Image processing
opencv-python>=4.8.0
# Testing (optional)
pytest>=7.4.0
pytest-asyncio>=0.21.0