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:
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