fix: match v5 checkpoint architecture and benchmark scaled inference

5878871

verified

akashch1512 commited on 4 days ago