OPTIONAL COMMIT: Do not depend on zmq_control

Author: tdene

megatron/rl/inference/megatron.py

@@ -252,12 +252,12 @@ async def kill(self):
 
     async def suspend(self):
         if dist.get_rank() == 0:
-            await self._client.pause_engines()
-        await self._inference_engine.paused.wait()
+            self._client.pause_engines()
+        #await self._inference_engine.paused.wait()
 
     async def resume(self):
         if dist.get_rank() == 0:
             self._client.unpause_engines()
-        await self._inference_engine.running.wait()
+        #await self._inference_engine.running.wait()
 
 class MegatronChatLocal(ChatInferenceInterface, MegatronLocal): ...