File size: 683 Bytes
45e565b
a22f25e
 
 
 
45e565b
 
a22f25e
020aab8
a22f25e
45e565b
 
55606e3
45e565b
 
72e6f21
45e565b
 
 
 
 
 
b9d8080
45e565b
b9d8080
 
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
# Core ML packages
git+https://github.com/huggingface/diffusers
transformers>=4.44.0,<5.0.0
accelerate>=0.33.0,<1.0.0
sentencepiece>=0.2.0

# API and utilities
openai>=1.0.0,<2.0.0
Pillow>=10.0.0
requests>=2.31.0

# Gradio UI
gradio>=4.44.0

# GPU optimization
kernels>=0.3.0

# ZeroGPU AoTI support - REQUIRED for aoti_load/aoti_run functions
spaces>=0.40.1

# PyTorch with torch.export and AoTI support
# Note: torch>=2.4.0 required for torch.export, >=2.5.0 recommended for full AoTI
# torch is pre-installed in ZeroGPU environment

# Note: torchao removed due to version incompatibility with torch 2.8.0+
# FP8 quantization can be added later when torchao supports newer PyTorch