From 87c34e4dd4f546bc40ad567d2c4e3672e76e6fad Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?M=2E=20Yusuf=20Sar=C4=B1g=C3=B6z?= Date: Sun, 30 Jul 2023 01:09:22 +0300 Subject: [PATCH] gguf : update convert-llama-h5-to-gguf.py --- convert-llama-h5-to-gguf.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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: