import safetensors.torch
import os

path = "models/depth_anything_v2/model.safetensors"
temp_path = "models/depth_anything_v2/model_temp.safetensors"
try:
    tensors = safetensors.torch.load_file(path)
    safetensors.torch.save_file(tensors, temp_path, metadata={"format": "pt"})
    os.remove(path)
    os.rename(temp_path, path)
    print("Successfully patched model.safetensors")
except Exception as e:
    print("Error:", e)