def get_num_special_tokens(config: ml_collections.ConfigDict):
  """Gets the number of special tokens."""
  if not config.get('multi_task', False):
    return 1  # MASK
  num_slots = 16  # Placeholder for new tasks
  assert -min(task_registry.SpecialTokens) < num_slots
  return num_slots
Gets the number of special tokens.
videogvt/train_lib/maskgvt_train_lib.py