diff --git a/examples/rl_gsm8k/utils.py b/examples/rl_gsm8k/utils.py index 7f8c950c..3e17f113 100644 --- a/examples/rl_gsm8k/utils.py +++ b/examples/rl_gsm8k/utils.py @@ -216,11 +216,13 @@ def _cleanup(self) -> None: f.close() def __enter__(self) -> "VLLMServiceManager": - self._start_service() + #self._start_service() + self.ports = [8080, 8081] return self def __exit__(self, exc_type, exc_val, exc_tb) -> None: - self._cleanup() + return None + #self._cleanup() def get_stats(self): return self.stats