import torchfrom peft import PeftModelfrom transformers import AutoModelForCausalLM, AutoTokenizer # Load base model + LoRA adaptersbase_model = AutoModelForCausalLM.from_pretrained( "Qwen/Qwen3-8B", trust_remote_code=True, device_map="auto", torch_dtype=torch.bfloat16,)model = PeftModel.from_pretrained(base_model, "albertdatzira/qwen3-8b-htr-vines-lora")tokenizer = AutoTokenizer.from_pretrained("albertdatzira/qwen3-8b-htr-vines-lora") model.eval() # Correct HTR textsystem_prompt = ( "Ets un expert en post-correcció d'HTR dels diaris de Ramon Viñes. " "Corregeix els errors de reconeixement del text manuscrit. " "Només corregeix errors d'HTR evidents. No modifiquis l'ortografia " "de l'autor, noms propis, ni símbols especials (¶, ¬, ⟦ ⟧).") htr_text = "para el almuerzo (á mediddia). Me hristes son todos estos aniversarios" messages = [ {"role": "system", "content": system_prompt}, {"role": "user", "content": f"Corregeix el següent text HTR:\n{htr_text}"},] text = tokenizer.apply_chat_template( messages, tokenize=False, add_generation_prompt=True, enable_thinking=False,)inputs = tokenizer(text, return_tensors="pt").to(model.device) with torch.no_grad(): outputs = model.generate( **inputs, max_new_tokens=4096, do_sample=False, pad_token_id=tokenizer.pad_token_id, ) generated = outputs[0][inputs["input_ids"].shape[-1]:]corrected = tokenizer.decode(generated, skip_special_tokens=True)print(corrected)# -> "para el almuerzo (á mediodia). Qué tristes son todos estos aniversarios"