src/unlimited_ocr_rdna4/constants.py

from __future__ import annotations

MODEL_REPO = "baidu/Unlimited-OCR"
MODEL_REVISION = "07dea832e22aefee32ad281d4b80551282e1c168"
MODEL_WEIGHT_FILE = "model-00001-of-000001.safetensors"
MODEL_WEIGHT_BYTES = 6_672_547_120
MODEL_WEIGHT_SHA256 = "2bc48a7a110061ea58fff65d3169367eebe3aee371ca6968dc2219c1b2855fc6"
UPSTREAM_MODEL_CODE_SHA256 = "268bdcbe12cf37bf5a2debb53faf542e56570958a5d9f3314aab3cab2cf6cb48"
PATCHED_MODEL_CODE_SHA256 = "67d722bc082cf2524948ac75ba1cf014fa0bffe2bfbd2e59a9b8511d3654366c"

MODEL_MANIFEST_FILE = "rdna4-manifest.json"
MODEL_UPSTREAM_BACKUP_FILE = "modeling_unlimitedocr.py.upstream"

# Every behavior-defining file loaded by Transformers is pinned. The weight is
# listed separately because callers may choose whether to return its full hash,
# but verification always hashes it before trust_remote_code=True is used.
MODEL_PAYLOAD_SHA256 = {
    "LICENSE": "d985048c6d69429d685fdbe7557340aa0897c0fd8dc038299b148b9c75dc3383",
    "README.md": "4f573db255db5262bfe2ea92e459bfc11635a456f210e3a96289bb9605077a55",
    "config.json": "27246d03fd670904ec9601b1cb0861fbb79ec076830771daa8d943d6229946f9",
    "configuration_deepseek_v2.py": "b8470dd616ba8745fce6e27b093aef73a098863cc891b2477dcf9326a36000f7",
    "conversation.py": "ec7b6ce89bcda643de1f43269ffa66a7b2e65dc3ed30e427958f776546b4ba03",
    "deepencoder.py": "0ae2fb6d1e5ae8cf100fc32f854830acd08c821a0a1f23a94a76588c222ddcf2",
    MODEL_WEIGHT_FILE: MODEL_WEIGHT_SHA256,
    "model.safetensors.index.json": "354be1f2dcfb72ebb385e25465522ce5413a77c36f3b35fec088a3162a11af99",
    "modeling_deepseekv2.py": "74e36e6bd0ba7bc565ef76464a99baa8e6bccb710ae9c1007b54ac30b855fa4c",
    "modeling_unlimitedocr.py": PATCHED_MODEL_CODE_SHA256,
    MODEL_UPSTREAM_BACKUP_FILE: UPSTREAM_MODEL_CODE_SHA256,
    "processor_config.json": "92588cffb1d7032ec83d0a06c3a5171b41df5cbf432d68765441139a57899328",
    "special_tokens_map.json": "ab4bd57ce17d62e39e0a39e739de1e407484f090f0b2c7e391312bca7a5b061a",
    "tokenizer.json": "a02f8fd5228c90256bb4f6554c34a579d48f909e5beb232dc4afad870b55a8b4",
    "tokenizer_config.json": "a0cbe8464049da1f891b7a12676de06af4cb54c130995d42f71adc1c30c6e9f3",
}

MODEL_DOWNLOAD_SHA256 = {
    **MODEL_PAYLOAD_SHA256,
    "modeling_unlimitedocr.py": UPSTREAM_MODEL_CODE_SHA256,
}
MODEL_DOWNLOAD_SHA256.pop(MODEL_UPSTREAM_BACKUP_FILE)

SUPPORTED_ARCHITECTURES = frozenset({"gfx1200", "gfx1201"})
HARDWARE_VERIFIED_ARCHITECTURES = frozenset({"gfx1201"})
VERIFIED_TORCH_VERSION = "2.9.1+rocm7.2.1.gitff65f5bc"
VERIFIED_HIP_VERSION = "7.2.53211-e1a6bc5663"
VERIFIED_ROCM_VERSION = "7.2.1"
VERIFIED_PACKAGE_VERSIONS = {
    "numpy": "1.26.4",
    "pillow": "12.1.1",
    "pypdfium2": "5.12.1",
    "torch": "2.9.1+rocm7.2.1.lw.gitff65f5bc",
    "torchvision": "0.24.0+rocm7.2.1.gitb919bd0c",
    "transformers": "4.57.1",
    "triton": "3.5.1+rocm7.2.1.gita272dfa8",
}
VERSION = "0.1.0"