Update llmfoundry/models/inference_api_wrapper/openai_causal_lm.py

Co-authored-by: Daniel King <[email protected]>
mosaicml · Sep 15, 2023 · 2bd933a · 2bd933a
1 parent 1a9a77e
commit 2bd933a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmfoundry/models/inference_api_wrapper/openai_causal_lm.py b/llmfoundry/models/inference_api_wrapper/openai_causal_lm.py
@@ -120,7 +120,7 @@ def retokenize(self, tokens: List[int], cont_idxs: List[int]):  # noqa: D403
             self.tokenizer.decode(tokens[cont_idxs[0]:cont_idxs[-1] +
                                          1]).strip())['input_ids']
 
-        # replace the origina continuation with the retokenized continuation + padding
+        # replace the original continuation with the retokenized continuation + padding
         tokens = tokens[:cont_idxs[0]] + retokenized_continuation + [
             tokens[-1]
         ] * (len(tokens) -