Fix for Falcon error from PR #587

huggingface · Dec 20, 2023 · 230e9ad · 230e9ad
1 parent 21238af
commit 230e9ad
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/optimum/habana/utils.py b/optimum/habana/utils.py
@@ -340,7 +340,9 @@ def check_habana_frameworks_version(req_version):
     """
     Checks if the installed version of `habana_frameworks` is equal to `req_version`.
     """
-    return get_habana_frameworks_version() == version.parse(req_version)
+    return (
+        get_habana_frameworks_version().major == version.parse(req_version).major
+    ) and (get_habana_frameworks_version().minor == version.parse(req_version).minor)
 
 
 def get_device_name():