Fix maskformer model
removed in
!778 (comment 8714411)