Merge pull request GATEOverflow#104 from anandhu-eng/addChecksum

fixed variation name
anandhu-eng · Aug 21, 2024 · d08fd8f · d08fd8f
2 parents 267d661 + 6764d86
commit d08fd8f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/script/get-ml-model-bert-base-squad/_cm.json b/script/get-ml-model-bert-base-squad/_cm.json
@@ -62,7 +62,7 @@
       "deps": [
         {
           "names": [ "neural-magic-zoo-downloader" ],
-          "tags": "get,ml-model,zoo,deepsparse,_pruned95_obs_quant-none"
+          "tags": "get,ml-model,zoo,deepsparse,_bert-base-pruned95_obs_quant-none"
         }
       ],
       "env": {
@@ -87,6 +87,7 @@
     }
   },
   "print_env_at_the_end" : {
-    "CM_ML_MODEL_BERT_VOCAB_FILE_WITH_PATH": "Path to the BERT vocab file"
+    "CM_ML_MODEL_BERT_VOCAB_FILE_WITH_PATH": "Path to the BERT vocab file",
+    "CM_ML_MODEL_FILE_WITH_PATH": "Path to the ML model"
   }
 }