We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent aa59857 commit 9afb769Copy full SHA for 9afb769
src/forge/actors/reference_model.py
@@ -115,7 +115,11 @@ def __post_init__(self):
115
@endpoint
116
async def setup(self):
117
engine_config = {f.name: getattr(self, f.name) for f in fields(self)}
118
- self.engine = ForgeEngine(ForgeJobConfig(**engine_config))
+ 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)
123
self.engine.checkpointer.load()
124
self.model = self.engine.model_parts[0] # No pipeline parallelism yet
125
self.model.eval()
0 commit comments