add PP-DocBee-7B weight (PaddlePaddle#986)

LokeZhou · Jan 12, 2025 · ad9665d · ad9665d
1 parent 4460bac
commit ad9665d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/paddlemix/examples/ppdocbee/README.md b/paddlemix/examples/ppdocbee/README.md
@@ -6,9 +6,10 @@ PP-DocBee 是PaddleMIX团队自研的一款专注于文档理解的多模态大
 
 **本仓库支持的模型权重:**
 
-| Model              |
-|--------------------|
-| PaddleMIX/PPDocBee-2B-1129 |
+| Model              | 模型大小 | Huggingface 仓库地址 |
+|--------------------|----------|--------------------|
+| PaddleMIX/PPDocBee-2B-1129 | 2B | [PPDocBee-2B-1210](https://huggingface.co/PaddleMIX/PPDocBee-2B-1210) |
+| PaddleMIX/PPDocBee-7B-1210 | 7B | [PPDocBee-7B-1210](https://huggingface.co/PaddleMIX/PPDocBee-7B-1210) |
 
 
 ## 2 环境准备

diff --git a/paddlemix/examples/ppdocbee/ppdocbee_infer.py b/paddlemix/examples/ppdocbee/ppdocbee_infer.py
@@ -117,7 +117,7 @@ def main(args):
             output_text = processor.batch_decode(
                 generated_ids[0], skip_special_tokens=True, clean_up_tokenization_spaces=False
             )
-        print("output_text:\n", output_text)
+        print("output_text:\n", output_text[0])
 
 
 if __name__ == "__main__":