OmerShah commited on
Commit
97f16e3
·
verified ·
1 Parent(s): 61153f4

Upload 9 files

Browse files
README.md CHANGED
@@ -1,28 +1,121 @@
1
  ---
 
2
  tags:
3
- - text-to-image
4
- - lora
5
- - diffusers
6
- - template:diffusion-lora
7
- widget:
8
- - output:
9
- url: images/Screenshot 2025-08-14 230148.png
10
- text: '-'
11
- base_model: unsloth/gemma-3-270m-it-GGUF
12
- instance_prompt: null
13
- license: mit
14
  ---
15
- # MedGemma-270M
16
 
17
- <Gallery />
18
 
19
- ## Model description
20
 
21
- MedGemma-270M is a 270M-parameter Gemma 3 model fine-tuned with LoRA on the MIRIAD-4.4M (https:&#x2F;&#x2F;huggingface.co&#x2F;datasets&#x2F;miriad&#x2F;miriad-4.4M) medical Q&amp;A dataset.
22
- This model is designed for fast, domain-specialized inference on small GPUs and CPUs.
23
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
24
 
25
- ## Download model
26
 
27
 
28
- [Download](/OmerShah/MedGemma/tree/main) them in the Files & versions tab.
 
1
  ---
2
+ base_model: unsloth/gemma-3-270m-it
3
  tags:
4
+ - text-generation-inference
5
+ - transformers
6
+ - unsloth
7
+ - gemma3_text
8
+ license: apache-2.0
9
+ language:
10
+ - en
 
 
 
 
11
  ---
 
12
 
13
+ ### PharmaQA‑270M
14
 
15
+ <img src="banner.png" width="800"/>
16
 
17
+ **PharmaQA‑270M** is a compact, instruction-tuned language model for pharmacology and pharmacy domains. Based on **Gemma3 (270M parameters)**, it was fine-tuned using LoRA and merged into a single model checkpoint for easy deployment. This model is optimized for **educational and research use cases**, especially where compute constraints are present.
 
18
 
19
+ ---
20
+
21
+ ### Model Details
22
+
23
+ | Property | Value |
24
+ | ------------------ | ---------------------------------------------------------------------------------------- |
25
+ | Base Model | Google Gemma3 (270M parameters) |
26
+ | Fine-tuning Method | LoRA using [Unsloth](https://github.com/unslothai/unsloth) |
27
+ | Dataset Used | 25,000 Q\&A pairs from [MIRIAD-4.4M](https://huggingface.co/datasets/miriad/miriad-4.4M) |
28
+ | Epochs | 3 |
29
+ | Final Format | Merged (base + LoRA weights) |
30
+ | Model Size | 270M |
31
+ | License | ODC-BY v1.0 dataset license (non-commercial) |
32
+ | Author | [Mohamed Yasser](https://huggingface.co/yasserrmd) |
33
+
34
+ ---
35
+
36
+ ### ⚠️ Caution & Intended Use
37
+
38
+ * **Do not use this model for real-world medical diagnosis, treatment, or care decisions.**
39
+ * The model was trained on MIRIAD Q\&A pairs generated via LLMs from biomedical literature.
40
+ * MIRIAD and this model must be used for **educational, research, and academic exploration only**.
41
+ * This model inherits all **OpenAI and ODC-BY v1.0** usage limitations associated with the dataset.
42
+
43
+ ---
44
+
45
+ ### Performance Summary
46
+
47
+ From evaluation on 50 unseen pharma questions:
48
+
49
+ | Metric | Value |
50
+ | ------------------------ | --------------------------------------------------- |
51
+ | Average Answer Length | 40.3 words |
52
+ | Longest Answer | 95 words |
53
+ | Shortest Answer | 12 words |
54
+ | Empty / Short Responses | 0 |
55
+ | Clinical Accuracy | ✅ Consistent terminology |
56
+ | Depth in Short Responses | ⚠️ Limited |
57
+ | Best Use Case | Lightweight educational deployment (MCQs, tutoring) |
58
+
59
+ ---
60
+
61
+ ### Sample Inference Code
62
+
63
+ ```python
64
+ import torch
65
+ from transformers import AutoTokenizer, AutoModelForCausalLM
66
+
67
+ model_name = "yasserrmd/PharmaQA-270M"
68
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
69
+ model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
70
+ model.eval()
71
+
72
+ question = "What is the mechanism of action of metformin?"
73
+ messages = [{"role": "user", "content": f"Q: {question} A:"}]
74
+
75
+ inputs = tokenizer.apply_chat_template(
76
+ messages,
77
+ add_generation_prompt=True,
78
+ return_tensors="pt",
79
+ tokenize=True,
80
+ return_dict=True
81
+ ).to(model.device)
82
+
83
+ if "token_type_ids" in inputs:
84
+ del inputs["token_type_ids"]
85
+
86
+ with torch.no_grad():
87
+ outputs = model.generate(
88
+ **inputs,
89
+ max_new_tokens=128,
90
+ temperature=0.7,
91
+ top_p=0.95,
92
+ repetition_penalty=1.05
93
+ )
94
+
95
+ response = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
96
+ answer = response.split("A:")[-1].strip()
97
+
98
+ print("💊 Question:", question)
99
+ print("🧠 Answer:", answer)
100
+ ```
101
+
102
+ ---
103
+
104
+ ### License
105
+
106
+ * **Model**: Open for academic and non-commercial use
107
+ * **Dataset**: [MIRIAD-4.4M](https://huggingface.co/datasets/miriad/miriad-4.4M) under ODC-BY v1.0
108
+
109
+ ---
110
+
111
+ ### Acknowledgements
112
+
113
+ * MIRIAD creators for making the dataset openly accessible.
114
+ * [Unsloth](https://github.com/unslothai/unsloth) team for enabling fast LoRA tuning on small GPUs.
115
+ * Hugging Face and Google for Gemma3 base model.
116
+
117
+ ---
118
 
 
119
 
120
 
121
+ [<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)
added_tokens.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "<image_soft_token>": 262144
3
+ }
chat_template.jinja ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {{ bos_token }}
2
+ {%- if messages[0]['role'] == 'system' -%}
3
+ {%- if messages[0]['content'] is string -%}
4
+ {%- set first_user_prefix = messages[0]['content'] + '
5
+
6
+ ' -%}
7
+ {%- else -%}
8
+ {%- set first_user_prefix = messages[0]['content'][0]['text'] + '
9
+
10
+ ' -%}
11
+ {%- endif -%}
12
+ {%- set loop_messages = messages[1:] -%}
13
+ {%- else -%}
14
+ {%- set first_user_prefix = "" -%}
15
+ {%- set loop_messages = messages -%}
16
+ {%- endif -%}
17
+ {%- for message in loop_messages -%}
18
+ {%- if (message['role'] == 'user') != (loop.index0 % 2 == 0) -%}
19
+ {{ raise_exception("Conversation roles must alternate user/assistant/user/assistant/...") }}
20
+ {%- endif -%}
21
+ {%- if (message['role'] == 'assistant') -%}
22
+ {%- set role = "model" -%}
23
+ {%- else -%}
24
+ {%- set role = message['role'] -%}
25
+ {%- endif -%}
26
+ {{ '<start_of_turn>' + role + '
27
+ ' + (first_user_prefix if loop.first else "") }}
28
+ {%- if message['content'] is string -%}
29
+ {{ message['content'] | trim }}
30
+ {%- elif message['content'] is iterable -%}
31
+ {%- for item in message['content'] -%}
32
+ {%- if item['type'] == 'image' -%}
33
+ {{ '<start_of_image>' }}
34
+ {%- elif item['type'] == 'text' -%}
35
+ {{ item['text'] | trim }}
36
+ {%- endif -%}
37
+ {%- endfor -%}
38
+ {%- else -%}
39
+ {{ raise_exception("Invalid content type") }}
40
+ {%- endif -%}
41
+ {{ '<end_of_turn>
42
+ ' }}
43
+ {%- endfor -%}
44
+ {%- if add_generation_prompt -%}
45
+ {{ '<start_of_turn>model
46
+ ' }}
47
+ {%- endif -%}
config.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_sliding_window_pattern": 6,
3
+ "architectures": [
4
+ "Gemma3ForCausalLM"
5
+ ],
6
+ "attention_bias": false,
7
+ "attention_dropout": 0.0,
8
+ "attn_logit_softcapping": null,
9
+ "bos_token_id": 2,
10
+ "eos_token_id": 106,
11
+ "final_logit_softcapping": null,
12
+ "head_dim": 256,
13
+ "hidden_activation": "gelu_pytorch_tanh",
14
+ "hidden_size": 640,
15
+ "initializer_range": 0.02,
16
+ "intermediate_size": 2048,
17
+ "layer_types": [
18
+ "sliding_attention",
19
+ "sliding_attention",
20
+ "sliding_attention",
21
+ "sliding_attention",
22
+ "sliding_attention",
23
+ "full_attention",
24
+ "sliding_attention",
25
+ "sliding_attention",
26
+ "sliding_attention",
27
+ "sliding_attention",
28
+ "sliding_attention",
29
+ "full_attention",
30
+ "sliding_attention",
31
+ "sliding_attention",
32
+ "sliding_attention",
33
+ "sliding_attention",
34
+ "sliding_attention",
35
+ "full_attention"
36
+ ],
37
+ "max_position_embeddings": 32768,
38
+ "model_type": "gemma3_text",
39
+ "num_attention_heads": 4,
40
+ "num_hidden_layers": 18,
41
+ "num_key_value_heads": 1,
42
+ "pad_token_id": 0,
43
+ "query_pre_attn_scalar": 256,
44
+ "rms_norm_eps": 1e-06,
45
+ "rope_local_base_freq": 10000.0,
46
+ "rope_scaling": null,
47
+ "rope_theta": 1000000.0,
48
+ "sliding_window": 512,
49
+ "torch_dtype": "float16",
50
+ "transformers_version": "4.55.0",
51
+ "unsloth_fixed": true,
52
+ "unsloth_version": "2025.8.5",
53
+ "use_bidirectional_attention": false,
54
+ "use_cache": true,
55
+ "vocab_size": 262144
56
+ }
generation_config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 2,
3
+ "cache_implementation": "hybrid",
4
+ "do_sample": true,
5
+ "eos_token_id": [
6
+ 1,
7
+ 106
8
+ ],
9
+ "max_length": 32768,
10
+ "pad_token_id": 0,
11
+ "top_k": 64,
12
+ "top_p": 0.95,
13
+ "transformers_version": "4.55.0"
14
+ }
gitattributes ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
37
+ Screenshot[[:space:]]2025-08-14[[:space:]]235319.png filter=lfs diff=lfs merge=lfs -text
38
+ banner.png filter=lfs diff=lfs merge=lfs -text
special_tokens_map.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "boi_token": "<start_of_image>",
3
+ "bos_token": {
4
+ "content": "<bos>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false
9
+ },
10
+ "eoi_token": "<end_of_image>",
11
+ "eos_token": {
12
+ "content": "<end_of_turn>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false
17
+ },
18
+ "image_token": "<image_soft_token>",
19
+ "pad_token": {
20
+ "content": "<pad>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false
25
+ },
26
+ "unk_token": {
27
+ "content": "<unk>",
28
+ "lstrip": false,
29
+ "normalized": false,
30
+ "rstrip": false,
31
+ "single_word": false
32
+ }
33
+ }
tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1299c11d7cf632ef3b4e11937501358ada021bbdf7c47638d13c0ee982f2e79c
3
+ size 4689074
tokenizer_config.json ADDED
The diff for this file is too large to render. See raw diff