Skip to content

Conversation

ilyalasy
Copy link
Contributor

When loading model converted from BatchTopK to JumpRelu, this error happens: AssertionError: JumpReLU threshold must be positive.

During conversion to jumprelu, jumprelu_threshold is set to 0 which contradicts with config validation rule.

I removed the validation as we expect to load log_threshold from checkpoint now.

@curt-tigges curt-tigges merged commit 89c903d into curt-tigges:main Jun 2, 2025
1 of 4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants