hamsinimk commited on
Commit
18e1515
·
verified ·
1 Parent(s): 2516548

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
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
- [More Information Needed]
 
 
 
 
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. -->