File size: 1,776 Bytes
20d8653 ee2f9c5 20d8653 ee2f9c5 20d8653 023b9ee 20d8653 023b9ee 20d8653 023b9ee 20d8653 023b9ee ee2f9c5 023b9ee ee2f9c5 023b9ee ee2f9c5 023b9ee ee2f9c5 20d8653 ee2f9c5 023b9ee ee2f9c5 20d8653 ee2f9c5 | 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 | ---
base_model: openai/gpt-oss-20b
datasets:
- FreedomIntelligence/medical-o1-verifiable-problem
library_name: transformers
model_name: gpt-oss-20b-medical-qa
tags:
- generated_from_trainer
- trl
- sft
licence: license
license: apache-2.0
language:
- en
pipeline_tag: text-generation
---
# Model Card for gpt-oss-20b-medical-qa
This model is a fine-tuned version of [openai/gpt-oss-20b](https://huggingface.co/openai/gpt-oss-20b) on the [FreedomIntelligence/medical-o1-verifiable-problem](https://huggingface.co/datasets/FreedomIntelligence/medical-o1-verifiable-problem) dataset.
It has been trained using [TRL](https://github.com/huggingface/trl).
## Quick start
```python
from transformers import pipeline
prompt = """<|start|>developer<|message|># Instructions
You are a medical expert with advanced knowledge in clinical reasoning and diagnostics. Respond with ONLY the final diagnosis/cause in ≤5 words.<|end|><|start|>user<|message|>An 88-year-old woman with osteoarthritis is experiencing mild epigastric discomfort and has vomited material resembling coffee grounds multiple times. Considering her use of naproxen, what is the most likely cause of her gastrointestinal blood loss?<|end|><|start|>assistant<|message|>"""
# Load pipeline
generator = pipeline(
"text-generation",
model="kingabzpro/gpt-oss-20b-medical-qa",
device="cuda" # or device=0
)
# Run inference (passing in chat-style format)
output = generator(
prompt,
max_new_tokens=20,
return_full_text=False
)[0]
print(output["generated_text"])
```
Output:
```bash
NSAID use
```
## Training procedure
This model was trained with SFT.
### Framework versions
- TRL: 0.21.0
- Transformers: 4.55.2
- Pytorch: 2.8.0.dev20250319+cu128
- Datasets: 4.0.0
- Tokenizers: 0.21.4 |