Unverified Commit 85447bb2 authored by jiqing-feng's avatar jiqing-feng Committed by GitHub
Browse files

rm useless condition since the previous condition contains it. (#25403)

parent 1564a81a
...@@ -1086,14 +1086,6 @@ class Trainer: ...@@ -1086,14 +1086,6 @@ class Trainer:
optimizer_kwargs.update(bnb_kwargs) optimizer_kwargs.update(bnb_kwargs)
except ImportError: except ImportError:
raise ValueError("Trainer tried to instantiate bnb optimizer but bnb is not installed!") raise ValueError("Trainer tried to instantiate bnb optimizer but bnb is not installed!")
elif args.optim == OptimizerNames.ADAMW_BNB:
try:
from bitsandbytes.optim import Adam8bit
optimizer_cls = Adam8bit
optimizer_kwargs.update(adam_kwargs)
except ImportError:
raise ValueError("Trainer tried to instantiate bnb Adam8bit but bnb is not installed!")
elif args.optim == OptimizerNames.ADAMW_ANYPRECISION: elif args.optim == OptimizerNames.ADAMW_ANYPRECISION:
try: try:
from torchdistx.optimizers import AnyPrecisionAdamW from torchdistx.optimizers import AnyPrecisionAdamW
......
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