Fix for Falcon error from PR #587 (#608)

* Fix for Falcon error from PR #587 * Reformatted
huggingface · Dec 21, 2023 · cba7f00 · cba7f00
1 parent 223a63c
commit cba7f00
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():