winglian commited on
Commit
c58034d
·
1 Parent(s): 28fd429

use pytorch 2.0.1

Browse files
.github/workflows/base.yml CHANGED
@@ -18,12 +18,12 @@ jobs:
18
  - cuda: "118"
19
  cuda_version: 11.8.0
20
  python_version: "3.9"
21
- pytorch: 2.0.0
22
  axolotl_extras:
23
  - cuda: "118"
24
  cuda_version: 11.8.0
25
  python_version: "3.10"
26
- pytorch: 2.0.0
27
  axolotl_extras:
28
  - cuda: "117"
29
  cuda_version: 11.7.1
@@ -33,7 +33,7 @@ jobs:
33
  - cuda: "118"
34
  cuda_version: 11.8.0
35
  python_version: "3.9"
36
- pytorch: 2.0.0
37
  axolotl_extras: gptq
38
  steps:
39
  - name: Checkout
 
18
  - cuda: "118"
19
  cuda_version: 11.8.0
20
  python_version: "3.9"
21
+ pytorch: 2.0.1
22
  axolotl_extras:
23
  - cuda: "118"
24
  cuda_version: 11.8.0
25
  python_version: "3.10"
26
+ pytorch: 2.0.1
27
  axolotl_extras:
28
  - cuda: "117"
29
  cuda_version: 11.7.1
 
33
  - cuda: "118"
34
  cuda_version: 11.8.0
35
  python_version: "3.9"
36
+ pytorch: 2.0.1
37
  axolotl_extras: gptq
38
  steps:
39
  - name: Checkout
.github/workflows/main.yml CHANGED
@@ -17,17 +17,17 @@ jobs:
17
  - cuda: cu118
18
  cuda_version: 11.8.0
19
  python_version: "3.9"
20
- pytorch: 2.0.0
21
  axolotl_extras:
22
  - cuda: cu118
23
  cuda_version: 11.8.0
24
  python_version: "3.10"
25
- pytorch: 2.0.0
26
  axolotl_extras:
27
  - cuda: cu118
28
  cuda_version: 11.8.0
29
  python_version: "3.9"
30
- pytorch: 2.0.0
31
  axolotl_extras: gptq
32
  - cuda: cu117
33
  cuda_version: 11.7.1
@@ -72,17 +72,17 @@ jobs:
72
  - cuda: cu118
73
  cuda_version: 11.8.0
74
  python_version: "3.9"
75
- pytorch: 2.0.0
76
  axolotl_extras:
77
  - cuda: cu118
78
  cuda_version: 11.8.0
79
  python_version: "3.10"
80
- pytorch: 2.0.0
81
  axolotl_extras:
82
  - cuda: cu118
83
  cuda_version: 11.8.0
84
  python_version: "3.9"
85
- pytorch: 2.0.0
86
  axolotl_extras: gptq
87
  - cuda: cu117
88
  cuda_version: 11.7.1
 
17
  - cuda: cu118
18
  cuda_version: 11.8.0
19
  python_version: "3.9"
20
+ pytorch: 2.0.1
21
  axolotl_extras:
22
  - cuda: cu118
23
  cuda_version: 11.8.0
24
  python_version: "3.10"
25
+ pytorch: 2.0.1
26
  axolotl_extras:
27
  - cuda: cu118
28
  cuda_version: 11.8.0
29
  python_version: "3.9"
30
+ pytorch: 2.0.1
31
  axolotl_extras: gptq
32
  - cuda: cu117
33
  cuda_version: 11.7.1
 
72
  - cuda: cu118
73
  cuda_version: 11.8.0
74
  python_version: "3.9"
75
+ pytorch: 2.0.1
76
  axolotl_extras:
77
  - cuda: cu118
78
  cuda_version: 11.8.0
79
  python_version: "3.10"
80
+ pytorch: 2.0.1
81
  axolotl_extras:
82
  - cuda: cu118
83
  cuda_version: 11.8.0
84
  python_version: "3.9"
85
+ pytorch: 2.0.1
86
  axolotl_extras: gptq
87
  - cuda: cu117
88
  cuda_version: 11.7.1