print debug only

94b6a2494b95bb22625edd2a2bc853b58186dc6e · 2024-07-16 01:28:30 -0700 · Alex Cheema

Files touched

M exo/api/chatgpt_api.py

Diff

commit 94b6a2494b95bb22625edd2a2bc853b58186dc6e
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Tue Jul 16 01:28:30 2024 -0700

    print debug only
---
 exo/api/chatgpt_api.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/exo/api/chatgpt_api.py b/exo/api/chatgpt_api.py
index 4200b10c..3dc2fb42 100644
--- a/exo/api/chatgpt_api.py
+++ b/exo/api/chatgpt_api.py
@@ -57,7 +57,6 @@ class ChatGPTAPI:
         timeout = 90
         start_time = time.time()
         while time.time() - start_time < timeout:
-            print("poll")
             try:
                 result, is_finished = await self.node.get_inference_result(request_id)
             except Exception as e:
@@ -96,7 +95,7 @@ class ChatGPTAPI:
         await runner.setup()
         site = web.TCPSite(runner, host, port)
         await site.start()
-        print(f"Starting ChatGPT API server at {host}:{port}")
+        if DEBUG >= 1: print(f"Starting ChatGPT API server at {host}:{port}")
 
 # Usage example
 if __name__ == "__main__":

← bf565f94 fix #7 no module named aiohttp · back to Exo · explain device equality in README d78d5b20 →