Skip to content

Commit d3e46b5

Browse files
committed
refactor final_steps
1 parent 251c9c9 commit d3e46b5

File tree

3 files changed

+268
-202
lines changed

3 files changed

+268
-202
lines changed

models/solar.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@ namespace pro
182182
}
183183

184184
ctx->move_to_layer(LayerAllocatorManager::Epilog);
185-
return final_steps(ctx, input_ids, hidden_states);
185+
return final_steps->forward(this, ctx, input_ids, hidden_states);
186186
}
187187

188188
void init_layer_fowarding(float alpha, int pair_num, const int *pairs)

0 commit comments

Comments
 (0)