Skip to content

Commit dac2c88

Browse files
committed
hardcode refmodel.checkpoint.folder to empty
1 parent 2b404c0 commit dac2c88

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
@@ -87,7 +87,11 @@ def __post_init__(self):
8787
@endpoint
8888
async def setup(self):
8989
engine_config = {f.name: getattr(self, f.name) for f in fields(self)}
90-
self.engine = ForgeEngine(ForgeJobConfig(**engine_config))
90+
engine_config = ForgeJobConfig(**engine_config)
91+
engine_config.checkpoint.folder = (
92+
"" # hardcode to empty to force load from initial_load_path
93+
)
94+
self.engine = ForgeEngine(engine_config)
9195
self.engine.checkpointer.load()
9296
self.model = self.engine.model_parts[0] # No pipeline parallelism yet
9397
self.model.eval()

0 commit comments

Comments
 (0)