Update README.md
Browse files
README.md
CHANGED
|
@@ -65,7 +65,11 @@ This is the model card of a 🤗 transformers model that has been pushed on the
|
|
| 65 |
|
| 66 |
<!-- This section is meant to convey both technical and sociotechnical limitations. -->
|
| 67 |
|
| 68 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 69 |
|
| 70 |
## Prompt Format
|
| 71 |
<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
|
|
|
|
| 65 |
|
| 66 |
<!-- This section is meant to convey both technical and sociotechnical limitations. -->
|
| 67 |
|
| 68 |
+
```{python}\n
|
| 69 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
|
| 70 |
+
tokenizer = AutoTokenizer.from_pretrained('hamsinimk/doctor_note_summarization_llm')
|
| 71 |
+
model = AutoModelForCausalLM.from_pretrained('hamsinimk/doctor_note_summarization_llm', device_map = 'auto', dtype = torch.bfloat16)
|
| 72 |
+
```
|
| 73 |
|
| 74 |
## Prompt Format
|
| 75 |
<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
|