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