Views
No views yet
pip install transformers torch accelerate1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model_path = "./qwen3_14b_sft_rl"
5tokenizer = AutoTokenizer.from_pretrained(model_path)
6model = AutoModelForCausalLM.from_pretrained(
7 model_path,
8 torch_dtype=torch.bfloat16,
9 device_map="auto"
10)1# Prepare input (paper error detection task)
2prompt = """Please detect errors in the following academic paper paragraph:
3
4[Paper content...]
5
6Please identify errors and provide correction suggestions."""
7
8# Encode input
9inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
10
11# Generate response
12with torch.no_grad():
13 outputs = model.generate(
14 **inputs,
15 max_new_tokens=512,
16 temperature=0.7,
17 do_sample=True,
18 pad_token_id=tokenizer.pad_token_id
19 )
20
21# Decode output
22response = tokenizer.decode(outputs[0], skip_special_tokens=True)
23print(response)