summaryrefslogtreecommitdiff
path: root/convert-lora-to-ggml.py
diff options
context:
space:
mode:
authorJared Van Bortel <jared@nomic.ai>2024-01-20 18:14:18 -0500
committerGitHub <noreply@github.com>2024-01-20 18:14:18 -0500
commitb43ebde3b0ccbc42d9dd782b32e2fd8eb35b43b5 (patch)
tree74759304364d6257681e092303c2e125b5113a6d /convert-lora-to-ggml.py
parent97c1549808d2742d37584a3c9df28154bdf34417 (diff)
convert : partially revert PR #4818 (#5041)
Diffstat (limited to 'convert-lora-to-ggml.py')
-rwxr-xr-xconvert-lora-to-ggml.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/convert-lora-to-ggml.py b/convert-lora-to-ggml.py
index 35ce152f..4904bf12 100755
--- a/convert-lora-to-ggml.py
+++ b/convert-lora-to-ggml.py
@@ -5,17 +5,16 @@ import json
import os
import struct
import sys
+from pathlib import Path
from typing import Any, BinaryIO, Sequence
import numpy as np
import torch
-from pathlib import Path
if 'NO_LOCAL_GGUF' not in os.environ:
sys.path.insert(1, str(Path(__file__).parent / 'gguf-py' / 'gguf'))
import gguf
-
NUMPY_TYPE_TO_FTYPE: dict[str, int] = {"float32": 0, "float16": 1}