File tree Expand file tree Collapse file tree 5 files changed +11
-11
lines changed Expand file tree Collapse file tree 5 files changed +11
-11
lines changed Original file line number Diff line number Diff line change @@ -178,7 +178,7 @@ jobs:
178
178
179
179
container :
180
180
image : diffusers/diffusers-pytorch-cuda
181
- options : --gpus 0 --shm-size "16gb" --ipc host
181
+ options : --gpus all --shm-size "16gb" --ipc host
182
182
183
183
steps :
184
184
- name : Checkout diffusers
@@ -344,7 +344,7 @@ jobs:
344
344
group : aws-g6e-xlarge-plus
345
345
container :
346
346
image : diffusers/diffusers-pytorch-cuda
347
- options : --shm-size "20gb" --ipc host --gpus 0
347
+ options : --shm-size "20gb" --ipc host --gpus all
348
348
steps :
349
349
- name : Checkout diffusers
350
350
uses : actions/checkout@v3
@@ -396,7 +396,7 @@ jobs:
396
396
group : aws-g6e-xlarge-plus
397
397
container :
398
398
image : diffusers/diffusers-pytorch-cuda
399
- options : --shm-size "20gb" --ipc host --gpus 0
399
+ options : --shm-size "20gb" --ipc host --gpus all
400
400
steps :
401
401
- name : Checkout diffusers
402
402
uses : actions/checkout@v3
Original file line number Diff line number Diff line change @@ -253,7 +253,7 @@ jobs:
253
253
254
254
container :
255
255
image : diffusers/diffusers-pytorch-cuda
256
- options : --gpus 0 --shm-size "16gb" --ipc host
256
+ options : --gpus all --shm-size "16gb" --ipc host
257
257
steps :
258
258
- name : Checkout diffusers
259
259
uses : actions/checkout@v3
Original file line number Diff line number Diff line change @@ -167,7 +167,7 @@ jobs:
167
167
168
168
container :
169
169
image : diffusers/diffusers-pytorch-cuda
170
- options : --gpus 0 --shm-size "16gb" --ipc host
170
+ options : --gpus all --shm-size "16gb" --ipc host
171
171
172
172
steps :
173
173
- name : Checkout diffusers
@@ -210,7 +210,7 @@ jobs:
210
210
211
211
container :
212
212
image : diffusers/diffusers-pytorch-xformers-cuda
213
- options : --gpus 0 --shm-size "16gb" --ipc host
213
+ options : --gpus all --shm-size "16gb" --ipc host
214
214
215
215
steps :
216
216
- name : Checkout diffusers
@@ -252,7 +252,7 @@ jobs:
252
252
253
253
container :
254
254
image : diffusers/diffusers-pytorch-cuda
255
- options : --gpus 0 --shm-size "16gb" --ipc host
255
+ options : --gpus all --shm-size "16gb" --ipc host
256
256
steps :
257
257
- name : Checkout diffusers
258
258
uses : actions/checkout@v3
Original file line number Diff line number Diff line change @@ -222,7 +222,7 @@ jobs:
222
222
223
223
container :
224
224
image : diffusers/diffusers-pytorch-cuda
225
- options : --gpus 0 --shm-size "16gb" --ipc host
225
+ options : --gpus all --shm-size "16gb" --ipc host
226
226
227
227
steps :
228
228
- name : Checkout diffusers
@@ -265,7 +265,7 @@ jobs:
265
265
266
266
container :
267
267
image : diffusers/diffusers-pytorch-xformers-cuda
268
- options : --gpus 0 --shm-size "16gb" --ipc host
268
+ options : --gpus all --shm-size "16gb" --ipc host
269
269
270
270
steps :
271
271
- name : Checkout diffusers
@@ -307,7 +307,7 @@ jobs:
307
307
308
308
container :
309
309
image : diffusers/diffusers-pytorch-cuda
310
- options : --gpus 0 --shm-size "16gb" --ipc host
310
+ options : --gpus all --shm-size "16gb" --ipc host
311
311
312
312
steps :
313
313
- name : Checkout diffusers
Original file line number Diff line number Diff line change 30
30
group : aws-g4dn-2xlarge
31
31
container :
32
32
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/
34
34
35
35
steps :
36
36
- name : Validate test files input
You can’t perform that action at this time.
0 commit comments