Hamid Shojanazeri 1 год назад
Родитель
Сommit
b2a55022cb
1 измененных файлов с 0 добавлено и 2 удалено
  1. 0 2
      llama_finetuning.py

+ 0 - 2
llama_finetuning.py

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