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