Fix shark_importer.py iree_utils import bug (huggingface#156)

nod-ai · Jun 23, 2022 · b7b2dfa · b7b2dfa
1 parent a7ca9b8
commit b7b2dfa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/shark/shark_importer.py b/shark/shark_importer.py
@@ -11,7 +11,7 @@
 import urllib.request
 from shark.shark_inference import SharkInference
 import iree.compiler.tflite as ireec_tflite
-from shark.iree_utils import IREE_TARGET_MAP
+from shark.iree_utils._common import IREE_TARGET_MAP
 
 
 class SharkImporter: