fix a bug where 'id2label' was incorrectly written as 'i2label' when …

…reading the config from pretrained config
huggingface · Nov 7, 2024 · 300c082 · 300c082
1 parent 7bbc624
commit 300c082
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/pytorch/image-classification/run_image_classification_no_trainer.py b/examples/pytorch/image-classification/run_image_classification_no_trainer.py
@@ -331,7 +331,7 @@ def main():
     config = AutoConfig.from_pretrained(
         args.model_name_or_path,
         num_labels=len(labels),
-        i2label=id2label,
+        id2label=id2label,
         label2id=label2id,
         finetuning_task="image-classification",
         trust_remote_code=args.trust_remote_code,