Views
No views yet
TRL, Transformers, PEFT.| Metric | Value | Description |
|---|---|---|
| BERTScore (F1) | 0.79 | High factual consistency and minimal hallucinations during paraphrasing. |
| Structure Adherence | 91% | Success rate in following the required format (bullet points, clear accents). |
inference.py script:1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3from peft import PeftModel
4
5model_id = "Qwen/Qwen2.5-7B-Instruct"
6adapter_id = "yuriy-magus/Qwen2.5-7B-Ecom-Refiner"
7device = "cuda" if torch.cuda.is_available() else "cpu"
8
9# 1. Load Tokenizer and Base Model
10tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
11model = AutoModelForCausalLM.from_pretrained(
12 model_id,
13 torch_dtype=torch.float16 if device == "cuda" else torch.float32,
14 device_map="auto" if device == "cuda" else None,
15 trust_remote_code=True
16)
17
18# 2. Load Adapter
19model = PeftModel.from_pretrained(model, adapter_id)
20model.to(device).eval()
21
22# 3. Prepare Prompt (as used in training)
23category = "Личные вещи / Одежда, обувь, аксессуары"
24title = "Кеды pataugas"
25original_desc = "Кеды фиpмы PATAUGAS. Кеды пoлнocтью кoжaные, пoдoшвa пpoшитa, мoлнии paбoчие. Сocтoяние хopoшее."
26prompt = f"""### Instruction:
27Отредактируй описание товара для Авито. Будь грамотным, добавь в текст структуру и привлекательность, а также строго придерживайся фактов из исходного текста.
28
29### Context:
30Категория: {category}
31Товар: {title}
32
33### Original Description:
34{original_desc}
35
36### Improved Description:
37"""
38
39# 4. Generate
40inputs = tokenizer(prompt, return_tensors="pt").to(device)
41with torch.no_grad():
42 outputs = model.generate(
43 **inputs,
44 max_new_tokens=200,
45 temperature=0.7,
46 top_p=0.9,
47 do_sample=True,
48 pad_token_id=tokenizer.eos_token_id
49 )
50
51result = tokenizer.decode(outputs[0], skip_special_tokens=True)
52print(result.split("### Improved Description:")[-1].strip())