markuz89's picture
MiniJEV v2 model export
026f63e verified
Raw History Blame Contribute Delete
204 Bytes
"""Model input names per state-encoder kind (torch-free, shared by training, runtime and ONNX)."""
STATE_INPUTS = {"grid": "state_grid", "text": "state_ids", "log": "state_log_ids", "numeric": "state_x"}