Upgrade to pytorch==2.2.0 (#217)

Author: tgaddair

Dockerfile

@@ -37,7 +37,7 @@ RUN cargo build --release
 # Adapted from: https://github.com/pytorch/pytorch/blob/master/Dockerfile
 FROM debian:bullseye-slim as pytorch-install
 
-ARG PYTORCH_VERSION=2.1.2
+ARG PYTORCH_VERSION=2.2.0
 ARG PYTHON_VERSION=3.10
 ARG CUDA_VERSION=11.8
 ARG MAMBA_VERSION=23.1.0-1

server/Makefile

@@ -18,7 +18,7 @@ gen-server:
 
 install: gen-server
 	pip install pip --upgrade
-	pip install torch==2.1.2
+	pip install torch==2.2.0
 	pip install -r requirements.txt
 	pip install -e ".[bnb, accelerate, quantize, peft]"