本模型由John Snow Labs开发。
本模型采用CC-BY-NC-ND许可协议,并须同时遵守本可接受使用政策。如您需要将本模型用于商业用途,请通过info@johnsnowlabs.com与我们联系。
!pip install -qU transformers accelerate
from transformers import AutoTokenizer
import transformers
import torch
model = "johnsnowlabs/JSL-MedLlama-3-8B-v2.0"
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"])| 任务 | 版本 | 过滤条件 | n-shot | 指标 | 数值 | 标准误差 | |
|---|---|---|---|---|---|---|---|
| stem | N/A | 无 | 0 | acc | 0.6466 | ± | 0.0056 |
| 无 | 0 | acc_norm | 0.6124 | ± | 0.0066 | ||
| - medmcqa | Yaml | 无 | 0 | acc | 0.6118 | ± | 0.0075 |
| 无 | 0 | acc_norm | 0.6118 | ± | 0.0075 | ||
| - medqa_4options | Yaml | 无 | 0 | acc | 0.6143 | ± | 0.0136 |
| 无 | 0 | acc_norm | 0.6143 | ± | 0.0136 | ||
| - anatomy (mmlu) | 0 | 无 | 0 | acc | 0.7185 | ± | 0.0389 |
| - clinical_knowledge (mmlu) | 0 | 无 | 0 | acc | 0.7811 | ± | 0.0254 |
| - college_biology (mmlu) | 0 | 无 | 0 | acc | 0.8264 | ± | 0.0317 |
| - college_medicine (mmlu) | 0 | 无 | 0 | acc | 0.7110 | ± | 0.0346 |
| - medical_genetics (mmlu) | 0 | 无 | 0 | acc | 0.8300 | ± | 0.0378 |
| - professional_medicine (mmlu) | 0 | 无 | 0 | acc | 0.7868 | ± | 0.0249 |
| - pubmedqa | 1 | 无 | 0 | acc | 0.7420 | ± | 0.0196 |
| 分组 | 版本 | 过滤条件 | n-shot | 指标 | 数值 | 标准误差 | |
|---|---|---|---|---|---|---|---|
| stem | N/A | 无 | 0 | acc | 0.6466 | ± | 0.0056 |
| 无 | 0 | acc_norm | 0.6124 | ± | 0.0066 |