Create README.md

README.md

@@ -0,0 +1,44 @@
+---
+language:
+- zh
+library_name: transformers
+pipeline_tag: text2text-generation
+---
+
+```python
+
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained("svjack/gpt-dialogue")
+
+model = AutoModelForCausalLM.from_pretrained("svjack/gpt-dialogue")
+
+tokenizer.decode(
+model.generate(
+    tokenizer.encode(
+            "你认为今天的天气怎么样？", return_tensors="pt", add_special_tokens=True
+        ), 128,
+    num_beams=2,
+    top_p = 0.95,
+    top_k = 50,
+    repetition_penalty = 2.5,
+    length_penalty=1.0,
+    early_stopping=True,
+)[0],
+skip_special_tokens = True
+)
+
+'''
+'你 认 为 今 天 的 天 气 怎 么 样 ？
+我 认 为 这 会 很 冷 ， 但 是 春 天 是 如 此 美 丽 。
+您 最 喜 欢 哪 个 季 节 ？
+我 真 的 不 知 道 。 也 许 在 山 上 有 一 些 雪 松 石 。
+听 起 来 不 错 。 我 们 可 以 去 海 滩 吗 ？
+当 然 。 我 们 应 该 尽 快 去 那 里 。
+好 的 。 让 我 们 看 看 它 是 否 准 备 好 了 。
+现 在 下 雨 了 。
+我 们 要 去 哪 里 见 面 ？
+我 们'
+'''
+
+```