فهرست منبع

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(