vqgan needs to be float32

2022-07-07 08:53:27 -04:00
parent da62298f06
commit c199507a7a
4 changed files with 7 additions and 6 deletions
@@ -1,12 +1,12 @@
 build:
-  cuda: "11.3"
+  cuda: "11.5.1"
  gpu: true
-  python_version: "3.8"
+  python_version: "3.10"
  system_packages:
    - "libgl1-mesa-glx"
    - "libglib2.0-0"
  python_packages:
-    - "min-dalle==0.2.35"
+    - "min-dalle==0.2.36"
  run:
    - pip install torch==1.11.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html