How to use harpreetmann/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, "harpreetmann/stack_exc_multilabel_base_class_head")
How to use harpreetmann/stack_exc_multilabel_base_class_head with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("harpreetmann/stack_exc_multilabel_base_class_head", dtype="auto")
The community tab is the place to discuss and collaborate with the HF community!