We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1d32128 commit aa96333Copy full SHA for aa96333
1 file changed
src/art/local/backend.py
@@ -1439,7 +1439,7 @@ async def _experimental_fork_checkpoint(
1439
# _state is a cached_property that reads get_last_checkpoint_dir() on
1440
# first access; if it was accessed before the fork, it cached the base
1441
# model and will never pick up the forked weights.
1442
- service = await self._get_service(model)
+ service = await self._get_service(cast(TrainableModel, model))
1443
if hasattr(service, "_state") and "_state" in service.__dict__:
1444
del service.__dict__["_state"]
1445
if verbose:
0 commit comments