Set 'num_labels=1' while loading the base_model to avoid size mismatch error while loading the peft_model.

by skmalviya - opened Jul 13, 2024

Jul 13, 2024

base_model = AutoModelForSequenceClassification.from_pretrained(config.base_model_name_or_path, num_labels=1)

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment