chunk_content:  'classifier.bias']
  
  
  	Copied
  from transformers import AutoModelForImageClassification, TrainingArguments, Trainer
  
  model = AutoModelForImageClassification.from_pretrained(
      model_checkpoint,
      label2id=label2id,
      id2label=id2label,
      ignore_mismatched_sizes=True,  # provide this in case you're planning to fine-tune an already fine-tuned checkpoint
  )
  Before creating a PeftModel, you can check the number of trainable parameters in the 