mirror of
https://github.com/ggml-org/llama.cpp.git
synced 2025-11-09 10:17:06 +00:00
gguf : update convert-llama-h5-to-gguf.py
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user