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" |