diff --git a/src/diffusers/utils/import_utils.py b/src/diffusers/utils/import_utils.py index 64e3e54887f5..7ec1e60e9c26 100644 --- a/src/diffusers/utils/import_utils.py +++ b/src/diffusers/utils/import_utils.py @@ -708,7 +708,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)