Adding the option to not log

megatron/text_generation_server.py

@@ -122,7 +122,14 @@ class MegatronGenerate(Resource):
             if random_seed < 0: 
                 return "random_seed must be a positive integer"
 
+        no_log = False
+        if "no_log" in request.get_json():
+            no_log = request.get_json()["no_log"]
+            if not isinstance(no_log, bool):
+                return "no_log must be a boolean value"
+        
         with lock:  # Need to get lock to keep multiple threads from hitting code
+            if not no_log:
                 print("request IP: " + str(request.remote_addr))
                 print(json.dumps(request.get_json()),flush=True)
                 print("start time: ", datetime.datetime.now())