 a branch name, a tag name, a commit id, or any identifier
allowed by Git.   original_config_file (str, optional) —
The path to the original config file that was used to train the model. If not provided, the config file
will be inferred from the checkpoint file.   model_type (str, optional) —
The type of model to load. If not provided, the model type will be inferred from the checkpoint file.   image_size (int, optional) —
The size of the image