init commit

README.md

@@ -8,11 +8,11 @@ tags:
 - sentencepiece
 inference: true
 widget:
-- text: "情感分析任务：【房间还是比较舒适的,酒店服务良好】这篇文章的情感态度是什么？正面评价/负面评价"
+- text: "情感分析任务：【房间还是比较舒适的,酒店服务良好】这篇文章的情感态度是什么？正面/负面"
 - type: "text-generation"
 ---
 
-# Randeng-T5-784M-MultiTask-Chinese
+# Randeng-T5-77M-MultiTask-Chinese
 
 - Github: [Fengshenbang-LM](https://github.com/IDEA-CCNL/Fengshenbang-LM)
 - Docs: [Fengshenbang-Docs](https://fengshenbang-doc.readthedocs.io/)
@@ -62,7 +62,7 @@ model.resize_token_embeddings(len(tokenizer))
 model.eval()
 
 # tokenize
-text = "情感分析任务：【房间还是比较舒适的,酒店服务良好】这篇文章的情感态度是什么？正面评价/负面评价"
+text = "情感分析任务：【房间还是比较舒适的,酒店服务良好】这篇文章的情感态度是什么？正面/负面"
 encode_dict = tokenizer(text, max_length=512, padding='max_length',truncation=True)
 
 inputs = {
@@ -74,15 +74,14 @@ inputs = {
 logits = model.generate(
   input_ids = inputs['input_ids'],
   max_length=100, 
-  do_sample= True
-  # early_stopping=True,
+  early_stopping=True,
   )
 
 logits=logits[:,1:]
 predict_label = [tokenizer.decode(i,skip_special_tokens=True) for i in logits]
 print(predict_label)
 
-# model output: 正面评价
+# model output: 正面
 ```
 
 ## 引用 Citation