Views
No views yet
langdetect fallback.paged_adamw_32bit)lm-evaluation-harness.
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "Sepolian/qwen2.5-0.5b-sft-openorca"
4
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
7
8messages = [
9 {"role": "system", "content": "You are a helpful assistant."},
10 {"role": "user", "content": "Explain quantum computing in simple terms."},
11]
12
13text = tokenizer.apply_chat_template(
14 messages,
15 tokenize=False,
16 add_generation_prompt=True
17)
18
19model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
20
21generated_ids = model.generate(
22 model_inputs.input_ids,
23 max_new_tokens=512
24)
25
26generated_ids = [
27 output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
28]
29
30response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
31print(response)1@misc{openorca,
2 title = {OpenOrca: An Open Dataset of GPT Augmented FLAN Reasoning Traces},
3 author = {Mukherjee, Subhabrata and Mitra, Arindam and Skjellum, Ganesh and Catalyurek, Umit and Tielelman, Thomas and Monroy-Hernandez, Andres},
4 year = {2023},
5 publisher = {HuggingFace}
6}