hku_diffusion_dllm / reference /code /T3D /requirements.txt
Ouzhang's picture
Add files using upload-large-folder tool
13c5606 verified
Raw
History Blame Contribute Delete
314 Bytes
torch==2.6.0
torchvision==0.21.0
transformers==4.52.4
accelerate
deepspeed
datasets
evaluate
peft
safetensors
flashinfer-python>=0.2.12
xformers
triton>=3.0.0
tqdm
numpy
scipy
pandas
scikit-learn
sentencepiece
huggingface-hub
omegaconf
liger_kernel
termcolor
nest_asyncio
ipykernel
math_verify
wandb
tensorboard