diff --git a/convert-llama-h5-to-gguf.py b/convert-llama-h5-to-gguf.py index ba9e9f677..0ea5e21aa 100644 --- a/convert-llama-h5-to-gguf.py +++ b/convert-llama-h5-to-gguf.py @@ -5,13 +5,13 @@ import sys import struct import json import numpy as np -from typing import List +from typing import Any, List from pathlib import Path from transformers import AutoModelForCausalLM from sentencepiece import SentencePieceProcessor -NDArray: 'TypeAlias' = 'np.ndarray[Any, Any]' +NDArray = np.ndarray[Any, Any] def permute(weights: NDArray, n_head: int) -> NDArray: