(path: str, arr: np.ndarray, shape)
| 42 | |
| 43 | |
| 44 | def save_raw(path: str, arr: np.ndarray, shape) -> None: |
| 45 | os.makedirs(os.path.dirname(path) or ".", exist_ok=True) |
| 46 | with open(path + ".bin", "wb") as f: |
| 47 | f.write(arr.astype(np.float32, copy=False).tobytes()) |
| 48 | with open(path + ".shape", "w", encoding="utf-8") as f: |
| 49 | f.write(",".join(str(int(d)) for d in shape)) |
| 50 | |
| 51 | |
| 52 | def save_ggml_bd(path: str, x_bd: torch.Tensor) -> None: |
no outgoing calls
no test coverage detected