kvcache-ai/ktransformers
1
0
Fork 0
mirror of https://github.com/kvcache-ai/ktransformers.git synced 2026-04-20 14:29:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
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()