min-dalle-test/cog.yaml

build:
  cuda: "11.3"
  gpu: true
  python_version: "3.8"
  system_packages:
    - "libgl1-mesa-glx"
    - "libglib2.0-0"
  python_packages:
    - "min-dalle==0.2.35"
  run:
    - pip install torch==1.11.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html

predict: "replicate_predictor.py:ReplicatePredictor"
replicate demo 2022-06-29 18:50:10 +00:00			`build:`
properly limit input to 64 tokens 2022-07-06 02:14:19 +00:00			`cuda: "11.3"`
replicate demo 2022-06-29 18:50:10 +00:00			`gpu: true`
control top_k value 2022-07-05 21:23:05 +00:00			`python_version: "3.8"`
replicate demo 2022-06-29 18:50:10 +00:00			`system_packages:`
			`- "libgl1-mesa-glx"`
			`- "libglib2.0-0"`
			`python_packages:`
support bfloat16 2022-07-07 12:21:20 +00:00			`- "min-dalle==0.2.35"`
update cog.yaml 2022-07-03 16:20:03 +00:00			`run:`
support bfloat16 2022-07-07 12:21:20 +00:00			`- pip install torch==1.11.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html`
replicate demo 2022-06-29 18:50:10 +00:00
update replicate files 2022-07-05 09:47:35 +00:00			`predict: "replicate_predictor.py:ReplicatePredictor"`