Skip to content

Commit 4c22f78

Browse files
committed
Remove registration of Adafactor because it was added to torch 2.5
1 parent 2d55390 commit 4c22f78

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

mmengine/optim/optimizer/builder.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -160,15 +160,9 @@ def register_bitsandbytes_optimizers() -> List[str]:
160160

161161

162162
def register_transformers_optimizers():
163-
transformer_optimizers = []
164-
try:
165-
from transformers import Adafactor
166-
except ImportError:
167-
pass
168-
else:
169-
OPTIMIZERS.register_module(name='Adafactor', module=Adafactor)
170-
transformer_optimizers.append('Adafactor')
171-
return transformer_optimizers
163+
return [
164+
'Adafactor',
165+
]
172166

173167

174168
TRANSFORMERS_OPTIMIZERS = register_transformers_optimizers()

0 commit comments

Comments
 (0)