mirror of
https://github.com/huggingface/transformers.git
synced 2025-10-20 17:13:56 +08:00
@ -5907,7 +5907,7 @@ if is_torch_available():
|
||||
)
|
||||
optim_test_params.append(
|
||||
(
|
||||
TrainingArguments(optim=OptimizerNames.ADAMW_TORCH_8BIT, output_dir="None"),
|
||||
OptimizerNames.ADAMW_TORCH_8BIT,
|
||||
AdamW8bit,
|
||||
default_adam_kwargs,
|
||||
)
|
||||
|
Reference in New Issue
Block a user