diff --git a/src/diffusers/utils/import_utils.py b/src/diffusers/utils/import_utils.py index f7244e97b8..b64cecc412 100644 --- a/src/diffusers/utils/import_utils.py +++ b/src/diffusers/utils/import_utils.py @@ -99,6 +99,7 @@ if USE_TORCH in ENV_VARS_TRUE_AND_AUTO_VALUES and USE_TF not in ENV_VARS_TRUE_VA 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"