Skip to content

Commit 713665d

Browse files
committed
fix types
1 parent ee2b35c commit 713665d

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

convert_lora_to_gguf.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def parse_args() -> argparse.Namespace:
128128
map_name[tensor_name] = dest_name
129129

130130
# overwrite method
131-
def map_tensor_name(self, name: str) -> Iterator[tuple[str, Tensor]]:
131+
def map_tensor_name(self, name: str) -> str:
132132
return map_name[name]
133133

134134
# overwrite method
@@ -141,13 +141,13 @@ def extra_f16_tensors(self, name: str, new_name: str, bid: int | None, n_dims: i
141141
del name, new_name, bid, n_dims # unused
142142
return ftype != gguf.LlamaFileType.ALL_F32
143143

144-
model_instance._map_tensor_name = model_instance.map_tensor_name
144+
model_instance._map_tensor_name = model_instance.map_tensor_name # type: ignore
145145
model_instance.map_tensor_name = types.MethodType(map_tensor_name, model_instance)
146146

147-
model_instance._get_tensors = model_instance.get_tensors
147+
model_instance._get_tensors = model_instance.get_tensors # type: ignore
148148
model_instance.get_tensors = types.MethodType(get_tensors, model_instance)
149149

150-
model_instance._extra_f16_tensors = model_instance.extra_f16_tensors
150+
model_instance._extra_f16_tensors = model_instance.extra_f16_tensors # type: ignore
151151
model_instance.extra_f16_tensors = types.MethodType(extra_f16_tensors, model_instance)
152152

153153
model_instance.gguf_writer.add_quantization_version(gguf.GGML_QUANT_VERSION)

0 commit comments

Comments
 (0)