Diffuse_Image_Generator / requirements.txt
cyrus-spc's picture
Adding test
f73e37d
raw
history blame contribute delete
340 Bytes
# PyTorch with CUDA support (for GPU acceleration)
# Install from PyTorch official channel for CUDA 11.8 or 12.1
torch>=2.0.0
torchvision>=0.15.0
torchaudio>=2.0.0
# Diffusion models and related packages
diffusers
transformers
accelerate
scipy
safetensors
# Optional for better performance
xformers
# Gradio for web interface
gradio