Update README.md

README.md

@@ -34,8 +34,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
 tokenizer = AutoTokenizer.from_pretrained("zenos-gpt-j-6B-alpaca-evol-4bit")
 model = AutoModelForCausalLM.from_pretrained("zenos-gpt-j-6B-alpaca-evol-4bit")
 
-# Generate text
-prompt = 'A continuación hay una instrucción que describe una tarea. Escribe una respuesta que complete adecuadamente lo que se pide.\n\n### Instrucción:\nEscribe un poema breve usando cuatro estrofas\n\n### Respuesta:\n'
+# Generate text; watch out the padding between [INST] ... [/INST]
+prompt = '[INST] Escribe un poema breve usando cuatro versos [/INST]'
 
 inputs = tokenizer(prompt, return_tensors="pt")
 input_ids = inputs["input_ids"].to(model.device)