winglian commited on
Commit
e31a4eb
1 Parent(s): ca86014
.github/workflows/build-ggml-cpp-wheel.yml CHANGED
@@ -30,7 +30,7 @@ jobs:
30
  run: docker build . -t artifact-builder -f Dockerfile-llama-cpp-wheel
31
 
32
  - name: Build Docker image
33
- run: docker build . -t artifact-builder --build-arg LLAMA_CPP_VERSION=${{ steps.separate.outputs.prefix }} -f Dockerfile-llama-cpp-wheel
34
 
35
  - name: Copy GPU artifact from Docker container
36
  run: |
 
30
  run: docker build . -t artifact-builder -f Dockerfile-llama-cpp-wheel
31
 
32
  - name: Build Docker image
33
+ run: docker build . -t artifact-builder --build-arg CTRANSFORMERS_VERSION=${{ steps.separate.outputs.prefix }} -f Dockerfile-llama-cpp-wheel
34
 
35
  - name: Copy GPU artifact from Docker container
36
  run: |
Dockerfile-ggml-cpp-wheel CHANGED
@@ -1,6 +1,6 @@
1
  FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04
2
 
3
- ARG CTRANSFORMERS_VERSION="v0.2.5"
4
  ARG CMAKE_VERSION=3.26
5
  ARG CMAKE_VERSION_PATCH=3.26.3
6
  ARG CMAKE_OS=linux
 
1
  FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04
2
 
3
+ ARG CTRANSFORMERS_VERSION="0.2.5"
4
  ARG CMAKE_VERSION=3.26
5
  ARG CMAKE_VERSION_PATCH=3.26.3
6
  ARG CMAKE_OS=linux