Bump to latest pytorch nightly 2.1.0.dev20230904

Includes flash attention v2

Author: njhill

Dockerfile

@@ -3,7 +3,7 @@ ARG BASE_UBI_IMAGE_TAG=9.2-722.1692769367
 ARG PROTOC_VERSION=24.1
 #ARG PYTORCH_INDEX="https://download.pytorch.org/whl"
 ARG PYTORCH_INDEX="https://download.pytorch.org/whl/nightly"
-ARG PYTORCH_VERSION=2.1.0.dev20230822
+ARG PYTORCH_VERSION=2.1.0.dev20230904
 
 ## Base Layer ##################################################################
 FROM registry.access.redhat.com/ubi9/ubi:${BASE_UBI_IMAGE_TAG} as base