Closed tomweingarten closed 3 years ago
When reversible layers are on, the call to MoE fails because it gets the unexpected keyword "_reversible". Happy to provide a stack trace if that's helpful.
@tomweingarten should be fixed in the latest!
Looks great, thanks!
When reversible layers are on, the call to MoE fails because it gets the unexpected keyword "_reversible". Happy to provide a stack trace if that's helpful.