Skip to content

Commit e864f1a

Browse files
authored
Update model path for bagel-tiny-random
1 parent 4c7883f commit e864f1a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/workers/rollout/rollout_vllm/test_vllm_omni_bagel_generate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
from verl.workers.rollout.replica import DiffusionOutput, RolloutMode
3939
from verl.workers.rollout.vllm_rollout.vllm_omni_async_server import vLLMOmniHttpServer
4040

41-
MODEL_PATH = Path(os.path.expanduser("~/models/zhengyuansu/bagel-tiny-random"))
41+
MODEL_PATH = Path(os.path.expanduser("~/models/tiny-random/bagel"))
4242
STAGE_CONFIG = os.environ.get("BAGEL_STAGE_CONFIG", "") # TODO: Point to the location on the CI server by default
4343

4444
DEFAULT_PROMPT = (

0 commit comments

Comments
 (0)