diff --git a/apps/accelerate/chatllama/setup.py b/apps/accelerate/chatllama/setup.py index b7035c13..d3107339 100644 --- a/apps/accelerate/chatllama/setup.py +++ b/apps/accelerate/chatllama/setup.py @@ -11,6 +11,7 @@ "torch", "tqdm", "transformers", + "openai" ] this_directory = Path(__file__).parent