diff --git a/tests/test_evaluate.py b/tests/test_evaluate.py index 6a5b66d..6c72b13 100644 --- a/tests/test_evaluate.py +++ b/tests/test_evaluate.py @@ -61,6 +61,12 @@ def setUpClass(cls) -> None: shell=True ).stdout.decode('utf-8').strip()[:7] + inference_context_spec_dict = None + try: + inference_context_spec_dict = cls.inference_context.spec_dict() + except Exception as e: + logger.warning(f"Inference context spec dict failed: {e}") + cls.results = { "results": evaluate( cls.pipeline, @@ -76,7 +82,7 @@ def setUpClass(cls) -> None: "dataset_name": TEST_DATASET_NAME, "model_version": TEST_MODEL_VERSION, "whisperkittools_commit_hash": wkt_commit_hash, - "inference_context": cls.inference_context.spec_dict(), + "inference_context": inference_context_spec_dict, "model_repo_id": MODEL_REPO_ID } } diff --git a/whisperkit/android/utils.py b/whisperkit/android/utils.py index 3fd1f03..e28cb22 100644 --- a/whisperkit/android/utils.py +++ b/whisperkit/android/utils.py @@ -20,7 +20,7 @@ "tiny", "tiny.en", "base", "base.en", "small", "small.en" ] -# Reference: +# Reference: # https://app.aihub.qualcomm.com/docs/hub/compile_examples.html#compiling-pytorch-model-to-a-qnn-model-library TARGET_TABLE = {