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