From 2fd004992970377d932fe2d50847af382d2478ee Mon Sep 17 00:00:00 2001 From: baptiste Date: Mon, 10 Mar 2025 10:49:32 +0000 Subject: [PATCH] fix(gaudi): add default argument for the dockerfile --- Dockerfile_gaudi | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile_gaudi b/Dockerfile_gaudi index 6d37c6ae..c814e912 100644 --- a/Dockerfile_gaudi +++ b/Dockerfile_gaudi @@ -1,6 +1,6 @@ # Those arguments are required to build the image -ARG HABANA_VERSION -ARG PYTORCH_VERSION +ARG HABANA_VERSION=1.19.0 +ARG PYTORCH_VERSION=2.5.1 # Rust builder FROM lukemathwalker/cargo-chef:latest-rust-1.85.0 AS chef