UniBioTransfer / setup.sh
scy639's picture
Upload folder using huggingface_hub
2b534de verified
raw
history blame contribute delete
422 Bytes
# install correct version of torch and torchvision according to your cuda version
pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
pip install -r requirements.txt
pip install -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
pip install -e git+https://github.com/openai/CLIP.git@main#egg=clip