Просмотр исходного кода

removing the option to take prompt from cli

Hamid Shojanazeri 1 год назад
Родитель
Сommit
b18a186385
1 измененных файлов с 0 добавлено и 2 удалено
  1. 0 2
      inference/code-llama/code_infilling_example.py

+ 0 - 2
inference/code-llama/code_infilling_example.py

@@ -43,8 +43,6 @@ def main(
         ), f"Provided Prompt file does not exist {prompt_file}"
         ), f"Provided Prompt file does not exist {prompt_file}"
         with open(prompt_file, "r") as f:
         with open(prompt_file, "r") as f:
             user_prompt = f.read()
             user_prompt = f.read()
-    elif not sys.stdin.isatty():
-        user_prompt = "\n".join(sys.stdin.readlines())
     else:
     else:
         print("No user prompt provided. Exiting.")
         print("No user prompt provided. Exiting.")
         sys.exit(1)
         sys.exit(1)