浏览代码

Fix another test

Matthias Reso 3 月之前
父节点
当前提交
1e2226cdc4
共有 1 个文件被更改,包括 3 次插入3 次删除
  1. 3 3
      getting-started/inference/local_inference/chat_completion/chat_completion.py

+ 3 - 3
getting-started/inference/local_inference/chat_completion/chat_completion.py

@@ -11,9 +11,9 @@ import sys
 import torch
 from transformers import AutoTokenizer
 
-from llama_recipes.inference.chat_utils import read_dialogs_from_file
-from llama_recipes.inference.model_utils import load_model, load_peft_model
-from llama_recipes.inference.safety_utils import get_safety_checker
+from llama_cookbook.inference.chat_utils import read_dialogs_from_file
+from llama_cookbook.inference.model_utils import load_model, load_peft_model
+from llama_cookbook.inference.safety_utils import get_safety_checker
 from accelerate.utils import is_xpu_available
 
 def main(