vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

convert-*.py: dict_item --> Iterable

Browse source
This commit is contained in:
brian khuu 2024-07-14 11:24:25 +10:00
parent 455c0e53ac
commit 5ab1a84085
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
examples/convert_legacy_llama.py
View file

@ -25,7 +25,6 @@ from concurrent.futures import ProcessPoolExecutor, ThreadPoolExecutor
from dataclasses import dataclass
from pathlib import Path
from typing import TYPE_CHECKING, Any, Callable, IO, Iterable, Literal, TypeVar
from _collections_abc import dict_items
import numpy as np
@ -1043,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: dict_items[str, LazyTensor], expert_count:int | None) -> int:
def per_model_weight_count_estimation(tensors: Iterable[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: