mlnsio/text2sql

1# Configuration for Cog ⚙️2# Reference: https://github.com/replicate/cog/blob/main/docs/yaml.md3 4build:5  # set to true if your model requires a GPU6  gpu: true7 8  # a list of ubuntu apt packages to install9  # system_packages:10  #   - "libgl1-mesa-glx"11  #   - "libglib2.0-0"12 13  # python version in the form '3.11' or '3.11.4'14  python_version: "3.11.1"15 16  # a list of packages in the format <package-name>==<version>17  python_packages:18    - "transformers==4.36.2"19    - "torch==1.13.0"20    - "accelerate==0.25.0"21  #   - "torchvision==0.9.0"22 23  # commands run after the environment is setup24  # run:25  #   - "echo env is ready!"26  #   - "echo another command if needed"27 28# predict.py defines how predictions are run on your model29predict: "predict.py:Predictor"