From 5ddf7825ee8da6399f99a7d199d928feccd9de8d Mon Sep 17 00:00:00 2001
From: rasmi <rasmi@google.com>
Date: Mon, 2 Jun 2025 11:47:51 -0400
Subject: [PATCH] Set _torch_version to N/A if torch is disabled.
---
src/diffusers/utils/import_utils.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/src/diffusers/utils/import_utils.py b/src/diffusers/utils/import_utils.py
index f7244e97b878..b64cecc412ca 100644
--- a/src/diffusers/utils/import_utils.py
+++ b/src/diffusers/utils/import_utils.py
@@ -99,6 +99,7 @@ def _is_package_available(pkg_name: str, get_dist_name: bool = False) -> Tuple[b
else:
logger.info("Disabling PyTorch because USE_TORCH is set")
_torch_available = False
+ _torch_version = "N/A"
_jax_version = "N/A"
_flax_version = "N/A"