Skip to content

Commit 03ce53b

Browse files
committed
fixes
1 parent 5dfb439 commit 03ce53b

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

llama_cpp/llama_cpp.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
NewType,
1111
Optional,
1212
TYPE_CHECKING,
13+
List,
1314
)
1415

1516
from llama_cpp._ctypes_extensions import (
@@ -654,6 +655,12 @@ class llama_model_kv_override(ctypes.Structure):
654655
# ggml_backend_buffer_type_t buft;
655656
# };
656657

658+
class llama_model_tensor_buft_override(ctypes.Structure):
659+
_fields_ = [
660+
("pattern", ctypes.c_char_p),
661+
("buft", ctypes.c_int),
662+
]
663+
657664

658665
# struct llama_model_params {
659666
# // NULL-terminated list of devices to use for offloading (if NULL, all available devices are used)

0 commit comments

Comments
 (0)