diff --git a/tests/__init__.py b/tests/__init__.py
index 07f9da59..b47c1d0b 100644
--- 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,