Views
No views yet
bert-base-uncased1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2from peft import PeftModel
3
4# Load base model and tokenizer
5base_model_name = "bert-base-uncased"
6tokenizer = AutoTokenizer.from_pretrained(base_model_name)
7model = AutoModelForSequenceClassification.from_pretrained(base_model_name)
8
9# Load LoRA adapter
10lora_model = PeftModel.from_pretrained(model, "ALI-USER/bert-lora-newsgroups")
11
12# Inference
13text = "Hello world!"
14inputs = tokenizer(text, return_tensors="pt")
15outputs = lora_model(**inputs)
16logits = outputs.logits