Hamid Shojanazeri 1 рік тому
батько
коміт
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,