1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4model = AutoModelForCausalLM.from_pretrained("rajkumar/CardioMed-LLaMA3.2-1B", torch_dtype=torch.float16).cuda()
5tokenizer = AutoTokenizer.from_pretrained("rajkumar/CardioMed-LLaMA3.2-1B")
6
7prompt = """### Instruction:
8Provide an abstract and awareness information for the following disease: Myocardial Infarction
9
10### Response:
11"""
12
13inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
14outputs = model.generate(**inputs, max_new_tokens=512)
15print(tokenizer.decode(outputs[0], skip_special_tokens=True))
1### Instruction:
2Provide an abstract and awareness information for the following disease: Stroke
3
4### Response: