Views
No views yet
| Parameter | Value |
|---|---|
| LoRA Rank | 16 |
| LoRA Alpha | 32 |
| Dropout | 0.05 |
| Learning Rate | 1e-4 |
| Epochs | 3 |
| Max Sequence Length | 2048 |
1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4model_name = "tokhey/egyptian-mcq-generator-gemma-7b"
5
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7
8model = AutoModelForCausalLM.from_pretrained(
9 model_name,
10 torch_dtype=torch.float16,
11 device_map="auto",
12)
13
14messages = [
15 {
16 "role": "user",
17 "content": "Generate 10 official Egyptian Ministry English MCQs about Past Perfect. Return only valid JSON."
18 }
19]
20
21text = tokenizer.apply_chat_template(
22 messages,
23 tokenize=False,
24 add_generation_prompt=True,
25)
26
27inputs = tokenizer(text, return_tensors="pt").to(model.device)
28
29outputs = model.generate(
30 **inputs,
31 max_new_tokens=1500,
32 temperature=0.7,
33)
34
35print(
36 tokenizer.decode(
37 outputs[0][inputs.input_ids.shape[1]:],
38 skip_special_tokens=True,
39 )
40)1{
2 "statement": "...",
3 "correct_answer": "...",
4 "plausible_distractors": [
5 "...",
6 "...",
7 "..."
8 ],
9 "explanation": "..."
10}