revert change fixed by latest sparsezoo commit

neuralmagic · Sep 11, 2023 · d595952 · d595952
1 parent ba53908
commit d595952
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/deepsparse/transformers/helpers.py b/src/deepsparse/transformers/helpers.py
@@ -46,9 +46,9 @@
 _LOGGER = get_main_logger()
 
 _MODEL_DIR_ONNX_NAME = "model.onnx"
-_MODEL_DIR_CONFIG_NAME = "deployment/config.json"
-_MODEL_DIR_TOKENIZER_NAME = "deployment/tokenizer.json"
-_MODEL_DIR_TOKENIZER_CONFIG_NAME = "deployment/tokenizer_config.json"
+_MODEL_DIR_CONFIG_NAME = "config.json"
+_MODEL_DIR_TOKENIZER_NAME = "tokenizer.json"
+_MODEL_DIR_TOKENIZER_CONFIG_NAME = "tokenizer_config.json"
 
 
 def get_onnx_path(model_path: str) -> str:

diff --git a/tests/helpers.py b/tests/helpers.py
@@ -68,9 +68,9 @@ def predownload_stub(stub: str, copy_framework_files: bool = False) -> Model:
     model_path = model.path
     if copy_framework_files:
         # required for `deepsparse.transformers.run_inference` on local model files
-        config_path = model.deployment.default.get_file("deployment/config.json").path
+        config_path = model.deployment.default.get_file("config.json").path
         tokenizer_config_path = model.deployment.default.get_file(
-            "deployment/tokenizer.json"
+            "okenizer.json"
         ).path
         shutil.copy(config_path, model_path)
         shutil.copy(tokenizer_config_path, model_path)