File size: 3,084 Bytes
5e1e3e5
a2e6ab3
9ecee26
 
 
 
 
 
 
5e1e3e5
 
ea437d2
5e1e3e5
 
04fb1aa
5e1e3e5
8b33ea8
04fb1aa
 
67ec914
04fb1aa
ea437d2
 
 
 
 
5e1e3e5
ea437d2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5e1e3e5
ea437d2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
a2e6ab3
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
---
license: cc-by-nc-nd-4.0
language:
- en
library_name: transformers
tags:
- reward model
- RLHF
- medical
---

# JSL-MedMNX-7B


[<img src="https://repository-images.githubusercontent.com/104670986/2e728700-ace4-11ea-9cfc-f3e060b25ddf">](http://www.johnsnowlabs.com)


JSL-MedMNX-7B is a 7 Billion parameter model developed by [John Snow Labs](https://www.johnsnowlabs.com/).
This model is trained on medical datasets to provide state-of-the-art performance on biomedical benchmarks: [Open Medical LLM Leaderboard](https://huggingface.co/spaces/openlifescienceai/open_medical_llm_leaderboard).

This model is available under a [CC-BY-NC-ND](https://creativecommons.org/licenses/by-nc-nd/4.0/deed.en) license and must also conform to this [Acceptable Use Policy](https://huggingface.co/johnsnowlabs). If you need to license  this model for commercial use, please contact us at info@johnsnowlabs.com.

## 💻 Usage

```python
!pip install -qU transformers accelerate

from transformers import AutoTokenizer
import transformers
import torch

model = "johnsnowlabs/JSL-MedMNX-7B"
messages = [{"role": "user", "content": "What is a large language model?"}]

tokenizer = AutoTokenizer.from_pretrained(model)
prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
pipeline = transformers.pipeline(
    "text-generation",
    model=model,
    torch_dtype=torch.float16,
    device_map="auto",
)

outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
print(outputs[0]["generated_text"])
```
## 🏆 Evaluation

|             Tasks             |Version|Filter|n-shot| Metric |Value |   |Stderr|
|-------------------------------|-------|------|-----:|--------|-----:|---|-----:|
|stem                           |N/A    |none  |     0|acc_norm|0.5191|±  |0.0068|
|                               |       |none  |     0|acc     |0.5658|±  |0.0058|
| - medmcqa                     |Yaml   |none  |     0|acc     |0.5135|±  |0.0077|
|                               |       |none  |     0|acc_norm|0.5135|±  |0.0077|
| - medqa_4options              |Yaml   |none  |     0|acc     |0.5373|±  |0.0140|
|                               |       |none  |     0|acc_norm|0.5373|±  |0.0140|
| - anatomy (mmlu)              |      0|none  |     0|acc     |0.6370|±  |0.0415|
| - clinical_knowledge (mmlu)   |      0|none  |     0|acc     |0.7245|±  |0.0275|
| - college_biology (mmlu)      |      0|none  |     0|acc     |0.7500|±  |0.0362|
| - college_medicine (mmlu)     |      0|none  |     0|acc     |0.6590|±  |0.0361|
| - medical_genetics (mmlu)     |      0|none  |     0|acc     |0.7200|±  |0.0451|
| - professional_medicine (mmlu)|      0|none  |     0|acc     |0.7206|±  |0.0273|
| - pubmedqa                    |      1|none  |     0|acc     |0.7720|±  |0.0188|

|Groups|Version|Filter|n-shot| Metric |Value |   |Stderr|
|------|-------|------|-----:|--------|-----:|---|-----:|
|stem  |N/A    |none  |     0|acc_norm|0.5191|±  |0.0068|
|      |       |none  |     0|acc     |0.5658|±  |0.0058|