bigcode/the-stack-dedup
Viewer • Updated • 237M • 13.7k • 396
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:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("anthonyq7/aegis", dtype="auto")Base model
microsoft/codebert-base
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("anthonyq7/aegis", dtype="auto")