Simplified the model by always computing batch-first ff22f29 verified davda54 commited on 23 days ago