Abhilash Majumder 1 год назад
Родитель
Сommit
11465d6329
1 измененных файлов с 0 добавлено и 2 удалено
  1. 0 2
      src/llama_recipes/utils/train_utils.py

+ 0 - 2
src/llama_recipes/utils/train_utils.py

@@ -23,8 +23,6 @@ from llama_recipes.policies import fpSixteen,bfSixteen_mixed, get_llama_wrapper
 from llama_recipes.utils.memory_utils import MemoryTrace
 from accelerate.utils import is_xpu_available, is_ccl_available
 
-from accelerate.utils import is_xpu_available, is_ccl_available
-
 def set_tokenizer_params(tokenizer: LlamaTokenizer):
     tokenizer.pad_token_id = 0
     tokenizer.padding_side = "left"