Update README.md

README.md

@@ -11,7 +11,7 @@ import transformers as tf, torch
 model_name = "SpectraSuite/TriLM_3.9B_Unpacked"
 
 # Please adjust the temperature, repetition penalty, top_k, top_p and other sampling parameters according to your needs.
-pipeline = tf.pipeline("text-generation", model=model_id, model_kwargs={"torch_dtype": torch.float16}, device_map="auto")
+pipeline = tf.pipeline("text-generation", model=model_name, model_kwargs={"torch_dtype": torch.float16}, device_map="auto")
 
 # These are base (pretrained) LLMs that are not instruction and chat tuned. You may need to adjust your prompt accordingly.
 pipeline("Once upon a time")