Hi, I am curious how to set a variable to save the value of the "output_router_logits", thanks!
You can save it in the config: model.config.output_router_logits = True
model.config.output_router_logits = True
· Sign up or log in to comment