UniCalli_Dev / requirements.txt
Tianshuo-Xu
Move PyTorch nightly installation from runtime to build requirements.txt
9299898
--extra-index-url https://download.pytorch.org/whl/nightly/cu126
# Core PyTorch dependencies (installed during build)
torch --pre
torchvision --pre
torchao --pre
# HuggingFace ecosystem (pinned versions for compatibility)
transformers>=4.40,<5.0
huggingface-hub>=0.33.5,<1.0
git+https://github.com/huggingface/diffusers.git
accelerate
safetensors
sentencepiece
# Additional dependencies for calligraphy app
einops
timm
optimum-quanto
bitsandbytes
# Data processing
datasets
omegaconf
opencv-python-headless
matplotlib
onnxruntime
# Chinese text processing
pypinyin
# Web UI (spaces handles torch 2.8+ AOT compilation)
gradio>=5.0
spaces>=0.47.0
# Flash Attention 3 support (optional, for H100 GPUs)
kernels