Skip to content

Commit 5c41448

Browse files
committed
fixes
1 parent 50979fc commit 5c41448

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 (
@@ -660,6 +661,12 @@ class llama_model_kv_override(ctypes.Structure):
660661
# ggml_backend_buffer_type_t buft;
661662
# };
662663

664+
class llama_model_tensor_buft_override(ctypes.Structure):
665+
_fields_ = [
666+
("pattern", ctypes.c_char_p),
667+
("buft", ctypes.c_int),
668+
]
669+
663670

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

0 commit comments

Comments
 (0)