diff options
author | Bernat Vadell <hounter.caza@gmail.com> | 2023-03-17 10:47:06 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-03-17 10:47:06 +0100 |
commit | 2af23d30434a677c6416812eea52ccc0af65119c (patch) | |
tree | 900c5ac5ceef13f65194ca8334cde41ed3590c09 /convert-pth-to-ggml.py | |
parent | 904d2a8d6acd667c9633138d45a361d40fbf76d0 (diff) |
🚀 Dockerize llamacpp (#132)
* feat: dockerize llamacpp
* feat: split build & runtime stages
* split dockerfile into main & tools
* add quantize into tool docker image
* Update .devops/tools.sh
Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
* add docker action pipeline
* change CI to publish at github docker registry
* fix name runs-on macOS-latest is macos-latest (lowercase)
* include docker versioned images
* fix github action docker
* fix docker.yml
* feat: include all-in-one command tool & update readme.md
---------
Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
Diffstat (limited to 'convert-pth-to-ggml.py')
-rw-r--r-- | convert-pth-to-ggml.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/convert-pth-to-ggml.py b/convert-pth-to-ggml.py index 5c36e9c..d0eb213 100644 --- a/convert-pth-to-ggml.py +++ b/convert-pth-to-ggml.py @@ -16,7 +16,7 @@ # At the start of the ggml file we write the model parameters # and vocabulary. # - +import os import sys import json import struct @@ -64,6 +64,10 @@ if len(sys.argv) > 2: sys.exit(1) fname_out = sys.argv[1] + "/ggml-model-" + ftype_str[ftype] + ".bin" +if os.path.exists(fname_out): + print(f"Skip conversion, it already exists: {fname_out}") + sys.exit(0) + with open(fname_hparams, "r") as f: hparams = json.load(f) |