소스 검색

resolve merge conflict

abhilash1910 1 년 전
부모
커밋
820c68a047
1개의 변경된 파일1개의 추가작업 그리고 0개의 파일을 삭제
  1. 1 0
      examples/chat_completion/chat_completion.py

+ 1 - 0
examples/chat_completion/chat_completion.py

@@ -14,6 +14,7 @@ from llama_recipes.inference.chat_utils import read_dialogs_from_file, format_to
 from llama_recipes.inference.model_utils import load_model, load_peft_model
 from llama_recipes.inference.safety_utils import get_safety_checker
 
+from accelerate.utils import is_xpu_available
 
 def main(
     model_name,