File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -178,7 +178,7 @@ jobs:
178178
179179 container :
180180 image : diffusers/diffusers-pytorch-cuda
181- options : --gpus 0 --shm-size "16gb" --ipc host
181+ options : --gpus all --shm-size "16gb" --ipc host
182182
183183 steps :
184184 - name : Checkout diffusers
@@ -344,7 +344,7 @@ jobs:
344344 group : aws-g6e-xlarge-plus
345345 container :
346346 image : diffusers/diffusers-pytorch-cuda
347- options : --shm-size "20gb" --ipc host --gpus 0
347+ options : --shm-size "20gb" --ipc host --gpus all
348348 steps :
349349 - name : Checkout diffusers
350350 uses : actions/checkout@v3
@@ -396,7 +396,7 @@ jobs:
396396 group : aws-g6e-xlarge-plus
397397 container :
398398 image : diffusers/diffusers-pytorch-cuda
399- options : --shm-size "20gb" --ipc host --gpus 0
399+ options : --shm-size "20gb" --ipc host --gpus all
400400 steps :
401401 - name : Checkout diffusers
402402 uses : actions/checkout@v3
Original file line number Diff line number Diff line change @@ -253,7 +253,7 @@ jobs:
253253
254254 container :
255255 image : diffusers/diffusers-pytorch-cuda
256- options : --gpus 0 --shm-size "16gb" --ipc host
256+ options : --gpus all --shm-size "16gb" --ipc host
257257 steps :
258258 - name : Checkout diffusers
259259 uses : actions/checkout@v3
Original file line number Diff line number Diff line change @@ -167,7 +167,7 @@ jobs:
167167
168168 container :
169169 image : diffusers/diffusers-pytorch-cuda
170- options : --gpus 0 --shm-size "16gb" --ipc host
170+ options : --gpus all --shm-size "16gb" --ipc host
171171
172172 steps :
173173 - name : Checkout diffusers
@@ -210,7 +210,7 @@ jobs:
210210
211211 container :
212212 image : diffusers/diffusers-pytorch-xformers-cuda
213- options : --gpus 0 --shm-size "16gb" --ipc host
213+ options : --gpus all --shm-size "16gb" --ipc host
214214
215215 steps :
216216 - name : Checkout diffusers
@@ -252,7 +252,7 @@ jobs:
252252
253253 container :
254254 image : diffusers/diffusers-pytorch-cuda
255- options : --gpus 0 --shm-size "16gb" --ipc host
255+ options : --gpus all --shm-size "16gb" --ipc host
256256 steps :
257257 - name : Checkout diffusers
258258 uses : actions/checkout@v3
Original file line number Diff line number Diff line change @@ -222,7 +222,7 @@ jobs:
222222
223223 container :
224224 image : diffusers/diffusers-pytorch-cuda
225- options : --gpus 0 --shm-size "16gb" --ipc host
225+ options : --gpus all --shm-size "16gb" --ipc host
226226
227227 steps :
228228 - name : Checkout diffusers
@@ -265,7 +265,7 @@ jobs:
265265
266266 container :
267267 image : diffusers/diffusers-pytorch-xformers-cuda
268- options : --gpus 0 --shm-size "16gb" --ipc host
268+ options : --gpus all --shm-size "16gb" --ipc host
269269
270270 steps :
271271 - name : Checkout diffusers
@@ -307,7 +307,7 @@ jobs:
307307
308308 container :
309309 image : diffusers/diffusers-pytorch-cuda
310- options : --gpus 0 --shm-size "16gb" --ipc host
310+ options : --gpus all --shm-size "16gb" --ipc host
311311
312312 steps :
313313 - name : Checkout diffusers
Original file line number Diff line number Diff line change 3030 group : aws-g4dn-2xlarge
3131 container :
3232 image : ${{ github.event.inputs.docker_image }}
33- options : --gpus 0 --privileged --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
33+ options : --gpus all --privileged --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/
3434
3535 steps :
3636 - name : Validate test files input
You can’t perform that action at this time.
0 commit comments