diff --git a/inference/code-llama/code_infilling_example.py b/inference/code-llama/code_infilling_example.py index 1111eb2b91bcac5a02448088c3f477640ee88a1b..601ea7c35df75bf87be9506c0bd98291e1c1cb63 100644 --- a/inference/code-llama/code_infilling_example.py +++ b/inference/code-llama/code_infilling_example.py @@ -43,8 +43,6 @@ def main( ), f"Provided Prompt file does not exist {prompt_file}" with open(prompt_file, "r") as f: user_prompt = f.read() - elif not sys.stdin.isatty(): - user_prompt = "\n".join(sys.stdin.readlines()) else: print("No user prompt provided. Exiting.") sys.exit(1)