"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "562b998366c7a4a2bd0addf1a860fbee0aa04d74"
Unverified Commit 86ffd5ff authored by Sourab Mangrulkar's avatar Sourab Mangrulkar Committed by GitHub
Browse files

fix name error when accelerate is not available (#26278)

* fix name error when accelerate is not available

* fix `is_fsdp_available`
parent 382ba670
......@@ -607,7 +607,7 @@ def is_accelerate_available(min_version: str = None):
def is_fsdp_available(min_version: str = "1.12.0"):
return version.parse(_torch_version) >= version.parse(min_version)
return is_torch_available() and version.parse(_torch_version) >= version.parse(min_version)
def is_optimum_available():
......
......@@ -79,6 +79,7 @@ if is_torch_available():
# hack to restore original logging level pre #21700
get_regression_trainer = partial(tests.trainer.test_trainer.get_regression_trainer, log_level="info")
require_fsdp_version = require_fsdp
if is_accelerate_available():
from accelerate.utils.constants import (
FSDP_PYTORCH_VERSION,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment