fix(encoder): use offline model in ci-base for pytorch transformer

gnes-ai · Aug 13, 2019 · c9e9272 · c9e9272
1 parent 5fedf6d
commit c9e9272
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -51,7 +51,7 @@ def line2pb_doc(line: str, doc_id: int = 0, deliminator: str = r'[.。！？!?]+
         'MOBILENET_MODEL': '/',
         'FASTERRCNN_MODEL': '/',
         'GNES_PROFILING': '',
-        'TORCH_TRANSFORMERS_MODEL': 'bert-base-uncased',
+        'TORCH_TRANSFORMERS_MODEL': '/torch_transformer'
     },
     'idc-165': {
         'BERT_CI_PORT': 7125,