use pytorch 2.0.1
Browse files- .github/workflows/base.yml +3 -3
- .github/workflows/main.yml +6 -6
    	
        .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. | 
| 22 | 
             
                        axolotl_extras:
         | 
| 23 | 
             
                      - cuda: "118"
         | 
| 24 | 
             
                        cuda_version: 11.8.0
         | 
| 25 | 
             
                        python_version: "3.10"
         | 
| 26 | 
            -
                        pytorch: 2.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. | 
| 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. | 
| 21 | 
             
                        axolotl_extras:
         | 
| 22 | 
             
                      - cuda: cu118
         | 
| 23 | 
             
                        cuda_version: 11.8.0
         | 
| 24 | 
             
                        python_version: "3.10"
         | 
| 25 | 
            -
                        pytorch: 2.0. | 
| 26 | 
             
                        axolotl_extras:
         | 
| 27 | 
             
                      - cuda: cu118
         | 
| 28 | 
             
                        cuda_version: 11.8.0
         | 
| 29 | 
             
                        python_version: "3.9"
         | 
| 30 | 
            -
                        pytorch: 2.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. | 
| 76 | 
             
                        axolotl_extras:
         | 
| 77 | 
             
                      - cuda: cu118
         | 
| 78 | 
             
                        cuda_version: 11.8.0
         | 
| 79 | 
             
                        python_version: "3.10"
         | 
| 80 | 
            -
                        pytorch: 2.0. | 
| 81 | 
             
                        axolotl_extras:
         | 
| 82 | 
             
                      - cuda: cu118
         | 
| 83 | 
             
                        cuda_version: 11.8.0
         | 
| 84 | 
             
                        python_version: "3.9"
         | 
| 85 | 
            -
                        pytorch: 2.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
         | 
