[gpu4] Simplify model code