Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3
4# Load the tokenizer from the base model
5base_model_path = "aisingapore/sea-lion-7b"
6tokenizer = AutoTokenizer.from_pretrained(base_model_path, trust_remote_code=True)
7
8# Load the fine-tuned model
9device = 'cuda' if torch.cuda.is_available() else 'cpu'
10peft_model_path = "mevsg/bntng-dis-v2"
11model = AutoModelForCausalLM.from_pretrained(peft_model_path, trust_remote_code=True)
12model.to(device)
13
14# Example usage
15prompt = "Apakah perkataan yang sesuai menggantikan 'bntng' dalam ayat berikut:"
16input_text = "Langit malam ini penuh dengan bntng yang bersinar terang."
17full_prompt = f"### USER:\n{prompt} {input_text}\n\n### RESPONSE:\n"
18
19tokens = tokenizer(full_prompt, return_tensors="pt")
20tokens.to(device)
21output = model.generate(
22 tokens["input_ids"],
23 attention_mask=tokens["attention_mask"],
24 max_new_tokens=20,
25 eos_token_id=tokenizer.eos_token_id)
26print(tokenizer.decode(output[0], skip_special_tokens=True))
27