import torchfrom transformers import AutoTokenizer, AutoModelForCausalLMfrom peft import PeftModel base_name = "Qwen/Qwen2.5-0.5B-Instruct"lora_repo = "ArtemMspektr/Laya-Med-Qwen2.5-0.5B-LoRA" tokenizer = AutoTokenizer.from_pretrained(lora_repo)base_model = AutoModelForCausalLM.from_pretrained(base_name, dtype=torch.float32)model = PeftModel.from_pretrained(base_model, lora_repo)model.eval() prompt = ( "<laya_deterministic_guardrail>BLOCKED_HARD_STOP | eGFR 18 < 30 ml/min | OpenFDA Boxed Warning: Lactic Acidosis</laya_deterministic_guardrail>\n" "Пациент с СД2 и СКФ 18 мл/мин, назначен Метформин 1000 мг 2 раза в день. Проведи клинический аудит.") inputs = tokenizer(tokenizer.apply_chat_template([ {"role": "system", "content": "You are Laya-Med-LLM, an expert clinical pharmacologist."}, {"role": "user", "content": prompt}], tokenize=False, add_generation_prompt=True), return_tensors="pt") with torch.no_grad(): outputs = model.generate(**inputs, max_new_tokens=180, temperature=0.1) print(tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True))