Sfoglia il codice sorgente

add docbee models to huggingface models (#4210)

Tingquan Gao 5 mesi fa
parent
commit
3f3e78b0bc
1 ha cambiato i file con 3 aggiunte e 0 eliminazioni
  1. 3 0
      paddlex/inference/utils/official_models.py

+ 3 - 0
paddlex/inference/utils/official_models.py

@@ -383,6 +383,9 @@ HUGGINGFACE_MODELS = [
     "PicoDet-L_layout_3cls",
     "PicoDet-S_layout_17cls",
     "PicoDet-S_layout_3cls",
+    "PP-DocBee2-3B",
+    "PP-DocBee-2B",
+    "PP-DocBee-7B",
     "PP-DocBlockLayout",
     "PP-DocLayout-L",
     "PP-DocLayout-M",