Update README.md

README.md

@@ -195,7 +195,7 @@ model = AutoModelForCausalLM.from_pretrained(model_name,
                                              )
 tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=True)
 streamer = TextStreamer(tokenizer, skip_special_tokens=True)
-pipe = pipeline("text-generation", model=base_model, tokenizer=tokenizer, streamer=streamer)
+pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, streamer=streamer)
 
 with autocast():
   output_default = pipe(formatted_prompt, pad_token_id=50256, max_new_tokens=128)