aboutsummaryrefslogtreecommitdiff
path: root/convert-pth-to-ggml.py
diff options
context:
space:
mode:
authorBernat Vadell <hounter.caza@gmail.com>2023-03-17 10:47:06 +0100
committerGitHub <noreply@github.com>2023-03-17 10:47:06 +0100
commit2af23d30434a677c6416812eea52ccc0af65119c (patch)
tree900c5ac5ceef13f65194ca8334cde41ed3590c09 /convert-pth-to-ggml.py
parent904d2a8d6acd667c9633138d45a361d40fbf76d0 (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.py6
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)