Views
No views yet
Important: This repo contains LoRA adapter weights only. Load the base model separately, then apply this adapter.
1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM
3from peft import PeftModel
4
5base_id = "Qwen/Qwen3-4B-Instruct-2507"
6adapter_id = "RinnRinnmini/lora_structeval_t_qwen3_4b_sft_v4"
7
8tokenizer = AutoTokenizer.from_pretrained(base_id, trust_remote_code=True)
9model = AutoModelForCausalLM.from_pretrained(
10 base_id,
11 device_map="auto",
12 torch_dtype=torch.float16,
13 trust_remote_code=True,
14)
15
16model = PeftModel.from_pretrained(model, adapter_id)
17model.eval()
18
19prompt = "Convert this JSON to YAML. Return ONLY YAML.\n\nJSON:\n{\"a\": 1, \"b\": [2,3]}"
20inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
21
22with torch.no_grad():
23 out = model.generate(**inputs, max_new_tokens=256, do_sample=False)
24
25print(tokenizer.decode(out[0], skip_special_tokens=True))