From dac2c884d8236cb074d0d981cbd8fc910bd5b42e Mon Sep 17 00:00:00 2001 From: DNXie Date: Tue, 14 Oct 2025 14:29:14 -0700 Subject: [PATCH] hardcode refmodel.checkpoint.folder to empty --- src/forge/actors/reference_model.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/forge/actors/reference_model.py b/src/forge/actors/reference_model.py index 7ab9f9527..89abc4471 100644 --- a/src/forge/actors/reference_model.py +++ b/src/forge/actors/reference_model.py @@ -87,7 +87,11 @@ def __post_init__(self): @endpoint async def setup(self): engine_config = {f.name: getattr(self, f.name) for f in fields(self)} - self.engine = ForgeEngine(ForgeJobConfig(**engine_config)) + engine_config = ForgeJobConfig(**engine_config) + engine_config.checkpoint.folder = ( + "" # hardcode to empty to force load from initial_load_path + ) + self.engine = ForgeEngine(engine_config) self.engine.checkpointer.load() self.model = self.engine.model_parts[0] # No pipeline parallelism yet self.model.eval()