bigcode/the-stack-dedup
Viewer • Updated • 237M • 14.1k • 410
How to use anthonyq7/aegis with PEFT:
from peft import PeftModel
from transformers import AutoModelForSequenceClassification
base_model = AutoModelForSequenceClassification.from_pretrained("microsoft/codebert-base")
model = PeftModel.from_pretrained(base_model, "anthonyq7/aegis")How to use anthonyq7/aegis with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("anthonyq7/aegis", device_map="auto")Base model
microsoft/codebert-base