From 8629b7bdc2d47375462ab0d954421fefc9303a99 Mon Sep 17 00:00:00 2001 From: brian khuu Date: Sun, 14 Jul 2024 12:19:23 +1000 Subject: [PATCH] covert-*.py: per_model_weight_count_estimation() tensor arg type is Iterable[tuple[str, LazyTensor]] --- examples/convert_legacy_llama.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/convert_legacy_llama.py b/examples/convert_legacy_llama.py index 3b2727aed..fc8a08cdf 100755 --- a/examples/convert_legacy_llama.py +++ b/examples/convert_legacy_llama.py @@ -1042,7 +1042,7 @@ def pick_output_type(model: LazyModel, output_type_str: str | None) -> GGMLFileT raise ValueError(f"Unexpected combination of types: {name_to_type}") -def per_model_weight_count_estimation(tensors: Iterable[str, LazyTensor], expert_count:int | None) -> int: +def per_model_weight_count_estimation(tensors: Iterable[tuple[str, LazyTensor]], expert_count:int | None) -> int: # TODO: Ensure parameter count is accurate throughout various model type sum_weight_estimate: int = 0 for name, lazy_tensor in tensors: