vrr/kvcache-ai-ktransformers
2
0
Fork 0
mirror of https://github.com/kvcache-ai/ktransformers.git synced 2026-04-28 03:39:48 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Update merge_safetensor_gguf.py

Browse source
This commit is contained in:
RICHARDNAN 2025-09-08 15:34:30 +08:00 • committed by GitHub
parent ecdd3f383f
commit ba9e964fcd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
merge_tensors/merge_safetensor_gguf.py
View file

@ -6,7 +6,7 @@ import sys
# sys.path.insert(0, "/home/azure/ktransformers")
import argparse
import torch
from ktransformers.util.custom_gguf import GGUFLoader, translate_name_to_gguf
from ktransformers.util.custom_loader import GGUFLoader, translate_name_to_gguf
from safetensors import safe_open
from safetensors.torch import save_file
import re
@ -211,4 +211,4 @@ def main():
return
if __name__ == "__main__":
main()
main()