Views
No views yet
htigenai/reflect_one: 16-bit version for higher precisionhtigenai/reflect_one_4bit: 4-bit quantized version for efficient deployment1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model_name = "htigenai/reflect_one" # or "htigenai/reflect_one_4bit"
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForCausalLM.from_pretrained(
7 model_name,
8 device_map="auto",
9 torch_dtype=torch.float16
10)
11
12# Format input following Llama 3.1 template
13input_text = """<|begin_of_text|><|start_header_id|>system<|end_header_id|>
14You are a reflective instructor for ISBEP.<|eot_id|><|start_header_id|>user<|end_header_id|>
15[Student reflection here]<|eot_id|><|start_header_id|>assistant<|end_header_id|>"""
16
17# Generate feedback
18inputs = tokenizer(input_text, return_tensors="pt").to(model.device)
19outputs = model.generate(
20 **inputs,
21 max_new_tokens=512,
22 temperature=0.7,
23 do_sample=True
24)
25response = tokenizer.decode(outputs[0], skip_special_tokens=True)1@misc{reflect_one_2025,
2 title={ReflectOne: Fine-tuned Llama 3.1 for Educational Reflection Feedback},
3 author={Oliveira, M.J.B., Ruijten - Dodoiu, P.},
4 year={2025},
5 publisher={unpublished}
6}