huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-26 12:32:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(gaudi): add default argument for the dockerfile

Browse Source
This commit is contained in:
baptiste 2025-03-10 10:49:32 +00:00
parent 02fa6adb20
commit 2fd0049929
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Dockerfile_gaudi
View File

@ -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