Browse Source

Update imports in chat_completion

Matthias Reso 1 year ago
parent
commit
5ac5d992ac
1 changed files with 3 additions and 3 deletions
  1. 3 3
      src/llama_recipes/inference/chat_completion.py

+ 3 - 3
src/llama_recipes/inference/chat_completion.py

@@ -6,14 +6,14 @@
 import fire
 import os
 import sys
-from typing import List
 
 import torch
-from model_utils import load_model, load_peft_model
 from transformers import LlamaTokenizer
-from safety_utils import get_safety_checker
 
 from .chat_utils import read_dialogs_from_file, format_tokens
+from .model_utils import load_model, load_peft_model
+from .safety_utils import get_safety_checker
+
 
 def main(
     model_name,