Installation

Requirements

• python version >= 3.8
• torch >= 1.13
• torchvision
• torchaudio
• fire
• configparser
• pandas <= 1.5.3
• scikit-learn >= 0.24.2
• diffusers >= 0.16.1
• deepspeed >= 0.9.0
• peft >= 0.3.0
• datasets >= 2.12.0
• transformers >= 4.29.1

Install Pypi

pip3 install unitorch
pip3 install unitorch[deepspeed]

Install Source

pip3 install \
    "git+https://github.com/fuliucansheng/unitorch#egg=unitorch[deepspeed]"

Install Extension

UNITORCH_EXTENSIONS=NGRAM pip3 install \
    "git+https://github.com/fuliucansheng/unitorch"