control top_k value

2022-07-05 17:23:05 -04:00
parent d64acb484c
commit 89a125b4b9
4 changed files with 13 additions and 6 deletions
@@ -1,13 +1,13 @@
 build:
-  cuda: "11.5.1"
+  cuda: "11.0"
  gpu: true
-  python_version: "3.10"
+  python_version: "3.8"
  system_packages:
    - "libgl1-mesa-glx"
    - "libglib2.0-0"
  python_packages:
-    - "min-dalle==0.2.27"
+    - "min-dalle==0.2.28"
  run:
-    - pip install torch==1.12.0+cu116 -f https://download.pytorch.org/whl/torch_stable.html
+    - pip install torch==1.10.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html

 predict: "replicate_predictor.py:ReplicatePredictor"