Spaces:
Running
Running
| """Hold a valid model artifact open without loading an inference runtime.""" | |
| from __future__ import annotations | |
| import hashlib | |
| import mmap | |
| import shutil | |
| import sys | |
| import time | |
| from pathlib import Path | |
| def main() -> int: | |
| model, mode = Path(sys.argv[1]), sys.argv[2] | |
| if mode == "map": | |
| with ( | |
| model.open("rb") as handle, | |
| mmap.mmap(handle.fileno(), 0, access=mmap.ACCESS_READ) as mapped, | |
| ): | |
| mapped[:64] | |
| time.sleep(0.7) | |
| elif mode == "copy": | |
| with model.open("rb") as source, (Path(sys.argv[3])).open("wb") as destination: | |
| shutil.copyfileobj(source, destination, length=64 * 1024) | |
| time.sleep(0.7) | |
| elif mode == "hash": | |
| with model.open("rb") as handle: | |
| hashlib.sha256(handle.read(1 << 20)).hexdigest() | |
| time.sleep(0.7) | |
| else: | |
| with model.open("rb") as handle: | |
| handle.read(1 << 20) | |
| time.sleep(0.7) | |
| return 0 | |
| if __name__ == "__main__": | |
| raise SystemExit(main()) | |