File size: 446 Bytes
9b4f999 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
build:
python_version: "3.8"
gpu: true
python_packages:
- 'huggingface_hub==0.8.1'
- 'torch==1.7.1'
- 'torchvision==0.8.2'
- 'tokenizers==0.10.2'
- 'pyflakes==2.2.0'
- 'tqdm==4.46.0'
- 'pytorch-lightning==1.5'
- 'einops==0.4.1'
- 'omegaconf==2.2.2'
- 'git+https://github.com/openai/CLIP.git'
- 'numpy==1.23.1'
- 'pillow==9.2.0'
- 'python-dotenv==0.20.0'
predict: "predict.py:Predictor" |