|
@@ -24,8 +24,6 @@ from transformers import (
|
|
BitsAndBytesConfig
|
|
BitsAndBytesConfig
|
|
)
|
|
)
|
|
import torch.distributed as dist
|
|
import torch.distributed as dist
|
|
-from optimum.bettertransformer import BetterTransformer
|
|
|
|
-
|
|
|
|
# Unused imports removed
|
|
# Unused imports removed
|
|
from utils.train_utils import (
|
|
from utils.train_utils import (
|
|
set_tokenizer_params,
|
|
set_tokenizer_params,
|