From 8ba4cdcb63a073d14baeb8fe87692cc6e89ff495 Mon Sep 17 00:00:00 2001 From: ZHU QIHAO <18811325956@163.com> Date: Thu, 26 Oct 2023 18:43:01 +0800 Subject: [PATCH] Update README.md --- README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README.md b/README.md index 42ddd00..4167a3e 100644 --- a/README.md +++ b/README.md @@ -144,6 +144,8 @@ inputs = tokenizer.encode(input_text, return_tensors="pt").to(device) outputs = model.generate(**inputs, max_length=128) print(tokenizer.decode(outputs[0])) ``` +![Completion GIF](pictures/completion_demo.gif) + #### Chat Model Inference ```python from transformers import AutoTokenizer, AutoModelForCausalLM