Skip to content

Commit f4527eb

Browse files
author
Dylan Huang
committed
Merge branch 'main' into derekx/fireworks-tracing
# Conflicts: # eval_protocol/pytest/remote_rollout_processor.py
2 parents fd204bb + cf55313 commit f4527eb

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

eval_protocol/pytest/remote_rollout_processor.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,10 @@ def __init__(
4242
self._model_base_url = model_base_url
4343
if os.getenv("EP_REMOTE_ROLLOUT_PROCESSOR_BASE_URL"):
4444
self._remote_base_url = os.getenv("EP_REMOTE_ROLLOUT_PROCESSOR_BASE_URL")
45+
self._model_base_url = model_base_url
46+
_ep_model_base_url = os.getenv("EP_MODEL_BASE_URL")
47+
if _ep_model_base_url:
48+
self._model_base_url = _ep_model_base_url
4549
self._poll_interval = poll_interval
4650
self._timeout_seconds = timeout_seconds
4751
self._output_data_loader = output_data_loader

0 commit comments

Comments
 (0)