Views
No views yet
1from transformers import AutoModelForSequenceClassification
2from peft import PeftConfig, PeftModel
3from transformers import AutoTokenizer
4
5# Load peft config model
6config = PeftConfig.from_pretrained("DEVCamiloSepulveda/33-Qwen3SP-aptanastudio-titanium")
7
8# Load tokenizer and model
9tokenizer = AutoTokenizer.from_pretrained("DEVCamiloSepulveda/33-Qwen3SP-aptanastudio-titanium")
10base_model = AutoModelForSequenceClassification.from_pretrained(
11 config.base_model_name_or_path,
12 num_labels=1,
13 torch_dtype=torch.float16,
14 device_map='auto'
15)
16model = PeftModel.from_pretrained(base_model, "DEVCamiloSepulveda/33-Qwen3SP-aptanastudio-titanium")
17
18# Prepare input text
19text = "Your issue description here"
20inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=20, padding="max_length")
21
22# Get prediction
23outputs = model(**inputs)
24story_points = outputs.logits.item()