Limit File Upload Size for Vendors

#-- Limit File Upload Size for Vendors --#
function limit_file_upload_size_for_vendors( $limit ) {

  $user_id    = get_current_user_id();
  $user_meta  = get_userdata( $user_id );
  $user_roles = $user_meta->roles;

  if ( in_array( 'seller', $user_roles ) ) {
    $limit = 2 * 1000000; // 2MB
  }
  return $limit;
}

add_filter( 'upload_size_limit', 'limit_file_upload_size_for_vendors' );