Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3import torch
4
5base_model = AutoModelForCausalLM.from_pretrained(
6 "TinyLlama/TinyLlama-1.1B-Chat-v1.0",
7 torch_dtype=torch.float16,
8 device_map="auto"
9)
10
11model = PeftModel.from_pretrained(
12 base_model,
13 "christianmalonga/virtual-professor"
14)
15
16tokenizer = AutoTokenizer.from_pretrained(
17 "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
18)
19
20prompt = """### Instruction:
21What is photosynthesis?
22
23### Response:
24"""
25
26inputs = tokenizer(prompt, return_tensors="pt")
27outputs = model.generate(**inputs, max_new_tokens=200)
28print(tokenizer.decode(outputs[0], skip_special_tokens=True))### Instruction:
{question}
### Response:
{answer}1@misc{malonga2025virtualprofessor,
2 author = {Christian Malonga},
3 title = {Virtual Professor AI: A Fine-tuned LLM
4 for Educational Assistance},
5 year = {2025},
6 publisher = {HuggingFace},
7 url = {https://huggingface.co/christianmalonga/virtual-professor}
8}