Update README.md

README.md

@@ -55,14 +55,6 @@ model = AutoModelForCausalLM.from_pretrained("Writer/palmyra-base", torch_dtype=
 # the fast tokenizer currently does not work correctly
 tokenizer = AutoTokenizer.from_pretrained("Writer/palmyra-base", use_fast=False)
 
-prompt = "What is the color of a carrot?\nA:"
-
-input_ids = tokenizer(prompt, return_tensors="pt").input_ids.cuda()
-
-generated_ids = model.generate(input_ids)
-
-tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
-
 
 ```