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 2b404c0 commit dac2c88Copy full SHA for dac2c88
src/forge/actors/reference_model.py
@@ -87,7 +87,11 @@ def __post_init__(self):
87
@endpoint
88
async def setup(self):
89
engine_config = {f.name: getattr(self, f.name) for f in fields(self)}
90
- self.engine = ForgeEngine(ForgeJobConfig(**engine_config))
+ 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)
95
self.engine.checkpointer.load()
96
self.model = self.engine.model_parts[0] # No pipeline parallelism yet
97
self.model.eval()
0 commit comments