Bump torch to version 2.2.1

ashleykleynhans · Mar 13, 2024 · b3e4b53 · b3e4b53
1 parent a1002a6
commit b3e4b53
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -15,7 +15,7 @@
 * Python 3.10.12
 * [Text Generation Web UI](
  https://github.com/oobabooga/text-generation-webui)
-* Torch 2.2.0
+* Torch 2.2.1
 * xformers 0.0.24
 * Jupyter Lab
 * [runpodctl](https://github.com/runpod/runpodctl)

diff --git a/docker-bake.hcl b/docker-bake.hcl
@@ -20,7 +20,7 @@ target "default" {
  args = {
  RELEASE = "${RELEASE}"
  INDEX_URL = "https://download.pytorch.org/whl/cu${CU_VERSION}"
- TORCH_VERSION = "2.2.0+cu${CU_VERSION}"
+ TORCH_VERSION = "2.2.1+cu${CU_VERSION}"
  XFORMERS_VERSION = "0.0.24"
  OOBABOOGA_COMMIT = "1934cb61ef879815644277c01c7295acbae542d8"
  RUNPODCTL_VERSION = "v1.14.2"