Update README.md

README.md

@@ -8,6 +8,7 @@ tags:
 - reasoning
 - reinforcement-learning
 - mini-llm
+- text-generation-inference
 datasets:
 - openai/gsm8k
 - openai/openai_humaneval
@@ -108,4 +109,4 @@ with torch.no_grad():
         pad_token_id=tokenizer.eos_token_id
     )
 
-print(tokenizer.decode(outputs[0], skip_special_tokens=True))
+print(tokenizer.decode(outputs[0], skip_special_tokens=True))