File size: 305 Bytes
e86746e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# Core dependencies
torch>=2.0.0
transformers>=4.30.0
huggingface_hub>=0.16.0
safetensors>=0.3.0
diffusers>=0.20.0

# Inference
lightning>=2.0.0
ftfy

# Configuration
omegaconf

# Utilities
numpy

# Note: flash-attn is required but needs special installation
# See README.md for installation instructions