From ba9f23be231cdcbaec2be8bc1338ba52ffb49f91 Mon Sep 17 00:00:00 2001 From: Eli Davidson Date: Fri, 31 Jan 2025 13:47:19 -0500 Subject: [PATCH] bump version of transformers and ubuntu for huggingface-pytorch-inference (#4524) Co-authored-by: Eli Davidson --- release_images_inference.yml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/release_images_inference.yml b/release_images_inference.yml index 81b15a099214..ccc038724385 100644 --- a/release_images_inference.yml +++ b/release_images_inference.yml @@ -149,11 +149,11 @@ release_images: framework: "huggingface_pytorch" version: "2.3.0" arch_type: "x86" - hf_transformers: "4.46.1" + hf_transformers: "4.48.0" inference: device_types: ["gpu" ] python_versions: [ "py311" ] - os_version: "ubuntu20.04" + os_version: "ubuntu22.04" cuda_version: "cu121" example: False disable_sm_tag: True @@ -162,7 +162,7 @@ release_images: framework: "huggingface_pytorch" version: "2.3.0" arch_type: "x86" - hf_transformers: "4.46.1" + hf_transformers: "4.48.0" inference: device_types: [ "cpu" ] python_versions: [ "py311" ]