The forward pass should stop at the last required block, rather than forwarding through the entire model.