From a4d3a00d98554e45ebdae3601618e6663555a9ba Mon Sep 17 00:00:00 2001
From: regisss <15324346+regisss@users.noreply.github.com>
Date: Mon, 19 Feb 2024 10:19:23 +0100
Subject: [PATCH] Fix dependencies (#56)

---
 Dockerfile            | 4 ++--
 rust-toolchain.toml   | 2 +-
 server/pyproject.toml | 4 ++--
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 74aff6ca..afdcfc87 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,5 @@
 # Rust builder
-FROM lukemathwalker/cargo-chef:latest-rust-1.71 AS chef
+FROM lukemathwalker/cargo-chef:latest-rust-1.75 AS chef
 WORKDIR /usr/src
 
 FROM chef as planner
@@ -31,7 +31,7 @@ COPY launcher launcher
 RUN cargo build --release
 
 # Text Generation Inference base image
-FROM vault.habana.ai/gaudi-docker/1.13.0/ubuntu22.04/habanalabs/pytorch-installer-2.1.0:latest as base
+FROM vault.habana.ai/gaudi-docker/1.14.0/ubuntu22.04/habanalabs/pytorch-installer-2.1.1:latest as base
 
 # Text Generation Inference base env
 ENV HUGGINGFACE_HUB_CACHE=/data \
diff --git a/rust-toolchain.toml b/rust-toolchain.toml
index 2db1883c..c97063e8 100644
--- a/rust-toolchain.toml
+++ b/rust-toolchain.toml
@@ -1,3 +1,3 @@
 [toolchain]
-channel = "1.70.0"
+channel = "1.75.0"
 components = ["rustfmt", "clippy"]
\ No newline at end of file
diff --git a/server/pyproject.toml b/server/pyproject.toml
index 73c9b929..101e007e 100644
--- a/server/pyproject.toml
+++ b/server/pyproject.toml
@@ -25,8 +25,8 @@ sentencepiece = "^0.1.97"
 tokenizers = "^0.14.1"
 huggingface-hub = "^0.16.4"
 peft = "^0.4.0"
-deepspeed = { git = "https://github.com/HabanaAI/DeepSpeed.git", branch = "1.13.0" }
-optimum-habana =  { git = "https://github.com/huggingface/optimum-habana.git", branch = "main" }
+deepspeed = { git = "https://github.com/HabanaAI/DeepSpeed.git", branch = "1.14.0" }
+optimum-habana =  "1.10.0"
 
 [tool.poetry.group.dev.dependencies]
 grpcio-tools = "*"