Skip to content

Commit

Permalink
bump version of transformers and ubuntu for huggingface-pytorch-infer…
Browse files Browse the repository at this point in the history
…ence (#4524)

Co-authored-by: Eli Davidson <elleedee@amazon.com>
  • Loading branch information
e-davidson and Eli Davidson authored Jan 31, 2025
1 parent 327910c commit ba9f23b
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions release_images_inference.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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" ]
Expand Down

0 comments on commit ba9f23b

Please # to comment.