Skip to content

Commit 9afb769

Browse files
authored
hardcode refmodel.checkpoint.folder to empty (#414)
1 parent aa59857 commit 9afb769

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

src/forge/actors/reference_model.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,11 @@ def __post_init__(self):
115115
@endpoint
116116
async def setup(self):
117117
engine_config = {f.name: getattr(self, f.name) for f in fields(self)}
118-
self.engine = ForgeEngine(ForgeJobConfig(**engine_config))
118+
engine_config = ForgeJobConfig(**engine_config)
119+
engine_config.checkpoint.folder = (
120+
"" # hardcode to empty to force load from initial_load_path
121+
)
122+
self.engine = ForgeEngine(engine_config)
119123
self.engine.checkpointer.load()
120124
self.model = self.engine.model_parts[0] # No pipeline parallelism yet
121125
self.model.eval()

0 commit comments

Comments
 (0)