diff --git a/src/diffusers/utils/import_utils.py b/src/diffusers/utils/import_utils.py index 639ed59d6393..ca88c64951f8 100644 --- a/src/diffusers/utils/import_utils.py +++ b/src/diffusers/utils/import_utils.py @@ -713,7 +713,7 @@ def is_torch_xla_version(operation: str, version: str): version (`str`): A string version of torch_xla """ - if not is_torch_xla_available: + if not _torch_xla_available: return False return compare_versions(parse(_torch_xla_version), operation, version)