winglian commited on
Commit
e3d0374
1 Parent(s): fac4600

add py310 support from base image

Browse files
Files changed (1) hide show
  1. .github/workflows/main.yml +20 -4
.github/workflows/main.yml CHANGED
@@ -15,14 +15,22 @@ jobs:
15
  include:
16
  - cuda: cu118
17
  cuda_version: 11.8.0
 
18
  pytorch: 2.0.0
19
  axolotl_extras:
20
  - cuda: cu118
21
  cuda_version: 11.8.0
 
 
 
 
 
 
22
  pytorch: 2.0.0
23
  axolotl_extras: gptq
24
  - cuda: cu117
25
  cuda_version: 11.7.0
 
26
  pytorch: 1.13.1
27
  axolotl_extras:
28
  runs-on: self-hosted
@@ -46,10 +54,10 @@ jobs:
46
  with:
47
  context: .
48
  build-args: |
49
- BASE_TAG=${{ github.ref_name }}-base-${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }}
50
  file: ./docker/Dockerfile
51
  push: ${{ github.event_name != 'pull_request' }}
52
- tags: ${{ steps.metadata.outputs.tags }}-${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }}
53
  labels: ${{ steps.metadata.outputs.labels }}
54
  cache-from: type=gha
55
  cache-to: type=gha,mode=max
@@ -62,14 +70,22 @@ jobs:
62
  include:
63
  - cuda: cu118
64
  cuda_version: 11.8.0
 
 
 
 
 
 
65
  pytorch: 2.0.0
66
  axolotl_extras:
67
  - cuda: cu118
68
  cuda_version: 11.8.0
 
69
  pytorch: 2.0.0
70
  axolotl_extras: gptq
71
  - cuda: cu117
72
  cuda_version: 11.7.0
 
73
  pytorch: 1.13.1
74
  axolotl_extras:
75
  runs-on: self-hosted
@@ -93,10 +109,10 @@ jobs:
93
  with:
94
  context: .
95
  build-args: |
96
- BASE_TAG=${{ github.ref_name }}-${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }}
97
  file: ./docker/Dockerfile-runpod
98
  push: ${{ github.event_name != 'pull_request' }}
99
- tags: ${{ steps.metadata.outputs.tags }}-${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }}
100
  labels: ${{ steps.metadata.outputs.labels }}
101
  cache-from: type=gha
102
  cache-to: type=gha,mode=max
 
15
  include:
16
  - cuda: cu118
17
  cuda_version: 11.8.0
18
+ python_version: "3.9"
19
  pytorch: 2.0.0
20
  axolotl_extras:
21
  - cuda: cu118
22
  cuda_version: 11.8.0
23
+ python_version: "3.10"
24
+ pytorch: 2.0.0
25
+ axolotl_extras:
26
+ - cuda: cu118
27
+ cuda_version: 11.8.0
28
+ python_version: "3.9"
29
  pytorch: 2.0.0
30
  axolotl_extras: gptq
31
  - cuda: cu117
32
  cuda_version: 11.7.0
33
+ python_version: "3.9"
34
  pytorch: 1.13.1
35
  axolotl_extras:
36
  runs-on: self-hosted
 
54
  with:
55
  context: .
56
  build-args: |
57
+ BASE_TAG=${{ github.ref_name }}-base-py${{ matrix.python_version }}-${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }}
58
  file: ./docker/Dockerfile
59
  push: ${{ github.event_name != 'pull_request' }}
60
+ tags: ${{ steps.metadata.outputs.tags }}-py${{ matrix.python_version }}-${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }}
61
  labels: ${{ steps.metadata.outputs.labels }}
62
  cache-from: type=gha
63
  cache-to: type=gha,mode=max
 
70
  include:
71
  - cuda: cu118
72
  cuda_version: 11.8.0
73
+ python_version: "3.9"
74
+ pytorch: 2.0.0
75
+ axolotl_extras:
76
+ - cuda: cu118
77
+ cuda_version: 11.8.0
78
+ python_version: "3.10"
79
  pytorch: 2.0.0
80
  axolotl_extras:
81
  - cuda: cu118
82
  cuda_version: 11.8.0
83
+ python_version: "3.9"
84
  pytorch: 2.0.0
85
  axolotl_extras: gptq
86
  - cuda: cu117
87
  cuda_version: 11.7.0
88
+ python_version: "3.9"
89
  pytorch: 1.13.1
90
  axolotl_extras:
91
  runs-on: self-hosted
 
109
  with:
110
  context: .
111
  build-args: |
112
+ BASE_TAG=${{ github.ref_name }}-py${{ matrix.python_version }}-${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }}
113
  file: ./docker/Dockerfile-runpod
114
  push: ${{ github.event_name != 'pull_request' }}
115
+ tags: ${{ steps.metadata.outputs.tags }}-py${{ matrix.python_version }}-${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }}
116
  labels: ${{ steps.metadata.outputs.labels }}
117
  cache-from: type=gha
118
  cache-to: type=gha,mode=max