Solomon / mlx /scripts /score_parity.py
kelseyway's picture
Make MLX adapter-only and include the reproducible BF16 converter
5c0a4a8
Raw
History Blame Contribute Delete
542 Bytes
"""Score a frozen panel for CUDA parity without fitting temperatures."""
import argparse
from solomon_mlx import Solomon
from solomon_mlx.evaluation import score_panel
def main():
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument("--model", default="models/quality")
parser.add_argument("--panel", required=True)
parser.add_argument("--output", required=True)
args = parser.parse_args()
score_panel(Solomon.load(args.model), args.panel, args.output)
if __name__ == "__main__":
main()