File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -86,14 +86,14 @@ jobs:
8686 pip_install_extras : " [hf,dev-test]"
8787 runner : linux-amd64-gpu-h100-latest-2
8888
89- # #### Speculative Decoding Example Tests (requires 25.08 image) #####
89+ # #### Speculative Decoding Example Tests (requires 26.01 image) #####
9090 speculative-decoding-pr :
9191 needs : [check-file-changes, wait-checks]
9292 if : startsWith(github.ref, 'refs/heads/pull-request/') && needs.check-file-changes.outputs.any_changed == 'true'
9393 uses : ./.github/workflows/_example_tests_runner.yml
9494 secrets : inherit
9595 with :
96- docker_image : " nvcr.io/nvidia/pytorch:25.08 -py3"
96+ docker_image : " nvcr.io/nvidia/pytorch:26.01 -py3"
9797 example : speculative_decoding
9898 pip_install_extras : " [hf,dev-test]"
9999 runner : linux-amd64-gpu-l4-latest-1
@@ -103,7 +103,7 @@ jobs:
103103 uses : ./.github/workflows/_example_tests_runner.yml
104104 secrets : inherit
105105 with :
106- docker_image : " nvcr.io/nvidia/pytorch:25.08 -py3"
106+ docker_image : " nvcr.io/nvidia/pytorch:26.01 -py3"
107107 example : speculative_decoding
108108 pip_install_extras : " [hf,dev-test]"
109109 runner : linux-amd64-gpu-h100-latest-2
You can’t perform that action at this time.
0 commit comments