# ==============================================================================
# COPYRIGHT (C) 2025-2026 KONSTANTIN VLADIMIROVICH GRABKO. ALL RIGHTS RESERVED.
# PATENT PENDING | CMS MANHATTAN JIRACK TECHNOLOGY
# ==============================================================================
import os
import json
from safetensors import safe_open

MODEL_PATH = "./JiRackTernary_70B_converted"
files = [f for f in os.listdir(MODEL_PATH) if f.endswith(".safetensors")]
files.sort()

metadata = {}
weight_map = {}

print(f"🔍 Сканирую {len(files)} шардов...")

for filename in files:
    full_path = os.path.join(MODEL_PATH, filename)
    with safe_open(full_path, framework="pt") as f:
        for key in f.keys():
            weight_map[key] = filename

index_data = {
    "metadata": {"total_size": 0}, # Размер можно не указывать точно
    "weight_map": weight_map
}

with open(os.path.join(MODEL_PATH, "model.safetensors.index.json"), "w") as f:
    json.dump(index_data, f, indent=2)

print("✅ Файл model.safetensors.index.json успешно создан!")