mhenrichsen
/

context-aware-splitter-7b

Text Generation

text-generation-inference

Model card Files Files and versions

mhenrichsen commited on Sep 19, 2023

Commit

92395d9

·

1 Parent(s): e7b982f

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -22,8 +22,8 @@ It returns a dict with the keys:
 ```python
 from transformers import AutoTokenizer, TextStreamer, AutoModelForCausalLM
-model = AutoModelForCausalLM.from_pretrained("mhenrichsen/context-aware-splitter-1b")
-tokenizer = AutoTokenizer.from_pretrained("mhenrichsen/context-aware-splitter-1b")
 streamer = TextStreamer(tokenizer, skip_special_tokens=True)
 WORD_SPLIT_COUNT = 50

 ```python
 from transformers import AutoTokenizer, TextStreamer, AutoModelForCausalLM
+model = AutoModelForCausalLM.from_pretrained("mhenrichsen/context-aware-splitter-7b")
+tokenizer = AutoTokenizer.from_pretrained("mhenrichsen/context-aware-splitter-7b")
 streamer = TextStreamer(tokenizer, skip_special_tokens=True)
 WORD_SPLIT_COUNT = 50