Skip to content

Commit 8524429

Browse files
committed
fixes
1 parent f65fbe7 commit 8524429

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 (
@@ -649,6 +650,12 @@ class llama_model_kv_override(ctypes.Structure):
649650
# ggml_backend_buffer_type_t buft;
650651
# };
651652

653+
class llama_model_tensor_buft_override(ctypes.Structure):
654+
_fields_ = [
655+
("pattern", ctypes.c_char_p),
656+
("buft", ctypes.c_int),
657+
]
658+
652659

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

0 commit comments

Comments
 (0)