1717 uses : actions/checkout@v2
1818 - name : Pull image
1919 run : |
20- docker pull vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest
20+ docker pull vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest
2121 - name : Run tests
2222 run : |
2323 docker run \
3030 --cap-add=sys_nice \
3131 --net=host \
3232 --ipc=host \
33- vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest \
33+ vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest \
3434 /bin/bash tests/ci/example_diff_tests.sh
3535 stable-diffusion :
3636 name : Test Stable Diffusion
4343 uses : actions/checkout@v2
4444 - name : Pull image
4545 run : |
46- docker pull vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest
46+ docker pull vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest
4747 - name : Run tests
4848 run : |
4949 docker run \
5959 --cap-add=sys_nice \
6060 --net=host \
6161 --ipc=host \
62- vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest \
62+ vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest \
6363 /bin/bash tests/ci/slow_tests_diffusers.sh ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
6464 deepspeed :
6565 name : Test DeepSpeed models
7272 uses : actions/checkout@v2
7373 - name : Pull image
7474 run : |
75- docker pull vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest
75+ docker pull vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest
7676 - name : Run tests
7777 run : |
7878 docker run \
8888 --cap-add=sys_nice \
8989 --net=host \
9090 --ipc=host \
91- vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest \
91+ vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest \
9292 /bin/bash tests/ci/slow_tests_deepspeed.sh ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
9393 fsdp :
9494 name : Test FSDP models
@@ -101,7 +101,7 @@ jobs:
101101 uses : actions/checkout@v2
102102 - name : Pull image
103103 run : |
104- docker pull vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest
104+ docker pull vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest
105105 - name : Run tests
106106 run : |
107107 docker run \
@@ -117,7 +117,7 @@ jobs:
117117 --cap-add=sys_nice \
118118 --net=host \
119119 --ipc=host \
120- vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest \
120+ vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest \
121121 make slow_tests_fsdp TOKEN=${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
122122 multi-card :
123123 name : Test multi-card models
@@ -130,7 +130,7 @@ jobs:
130130 uses : actions/checkout@v2
131131 - name : Pull image
132132 run : |
133- docker pull vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest
133+ docker pull vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest
134134 - name : Run tests
135135 run : |
136136 docker run \
@@ -146,7 +146,7 @@ jobs:
146146 --cap-add=sys_nice \
147147 --net=host \
148148 --ipc=host \
149- vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest \
149+ vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest \
150150 /bin/bash tests/ci/slow_tests_8x.sh ${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
151151 single-card :
152152 name : Test single-card models
@@ -160,7 +160,7 @@ jobs:
160160 uses : actions/checkout@v2
161161 - name : Pull image
162162 run : |
163- docker pull vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest
163+ docker pull vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest
164164 - name : Run tests
165165 run : |
166166 docker run \
@@ -177,7 +177,7 @@ jobs:
177177 --cap-add=sys_nice \
178178 --net=host \
179179 --ipc=host \
180- vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest \
180+ vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest \
181181 /bin/bash tests/ci/slow_tests_1x.sh
182182 text-generation :
183183 name : Test text-generation example
@@ -192,7 +192,7 @@ jobs:
192192 uses : actions/checkout@v2
193193 - name : Pull image
194194 run : |
195- docker pull vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest
195+ docker pull vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest
196196 - name : Run tests
197197 run : |
198198 docker run \
@@ -208,7 +208,7 @@ jobs:
208208 --cap-add=sys_nice \
209209 --net=host \
210210 --ipc=host \
211- vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest \
211+ vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest \
212212 make slow_tests_text_generation_example TOKEN=${{ secrets.TEXT_GENERATION_CI_HUB_TOKEN }}
213213 trl :
214214 name : Test TRL integration
@@ -221,7 +221,7 @@ jobs:
221221 uses : actions/checkout@v2
222222 - name : Pull image
223223 run : |
224- docker pull vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest
224+ docker pull vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest
225225 - name : Run tests
226226 run : |
227227 docker run \
@@ -237,7 +237,7 @@ jobs:
237237 --cap-add=sys_nice \
238238 --net=host \
239239 --ipc=host \
240- vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest \
240+ vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest \
241241 /bin/bash tests/ci/slow_tests_trl.sh
242242 sentence-transformers :
243243 name : Test Sentence Transformers integration
@@ -258,7 +258,7 @@ jobs:
258258 path : sentence-transformers
259259 - name : Pull image
260260 run : |
261- docker pull vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest
261+ docker pull vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest
262262 - name : Run tests
263263 run : |
264264 docker run \
@@ -274,5 +274,5 @@ jobs:
274274 --cap-add=sys_nice \
275275 --net=host \
276276 --ipc=host \
277- vault.habana.ai/gaudi-docker/1.22 .0/ubuntu22.04/habanalabs/pytorch-installer-2.7.1 :latest \
277+ vault.habana.ai/gaudi-docker/1.23 .0/ubuntu22.04/habanalabs/pytorch-installer-2.9.0 :latest \
278278 /bin/bash optimum-habana/tests/ci/sentence_transformers.sh
0 commit comments