How to use immortalPi/stack_exc_multilabel_base_class_head with PEFT:
from peft import PeftModel from transformers import AutoModelForSequenceClassification base_model = AutoModelForSequenceClassification.from_pretrained("google/gemma-2-2b") model = PeftModel.from_pretrained(base_model, "immortalPi/stack_exc_multilabel_base_class_head")
How to use immortalPi/stack_exc_multilabel_base_class_head with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("immortalPi/stack_exc_multilabel_base_class_head", device_map="auto")
