refactor: move to flax nnx #14
Annotations
10 errors
Code Checks:
src/xlens/hooked_transformer.py#L6
Import "jaxtyping" could not be resolved (reportMissingImports)
|
Code Checks:
src/xlens/hooked_transformer.py#L75
Return type is unknown (reportUnknownParameterType)
|
Code Checks:
src/xlens/hooked_transformer.py#L77
Type of parameter "input_ids" is unknown (reportUnknownParameterType)
|
Code Checks:
src/xlens/hooked_transformer.py#L77
Unexpected token at end of expression
|
Code Checks:
src/xlens/hooked_transformer.py#L77
"batch" is not defined (reportUndefinedVariable)
|
Code Checks:
src/xlens/hooked_transformer.py#L79
Unexpected token at end of expression
|
Code Checks:
src/xlens/hooked_transformer.py#L79
"batch" is not defined (reportUndefinedVariable)
|
Code Checks:
src/xlens/hooked_transformer.py#L117
Return type is unknown (reportUnknownParameterType)
|
Code Checks:
src/xlens/hooked_transformer.py#L119
Type of parameter "input_ids" is unknown (reportUnknownParameterType)
|
Code Checks:
src/xlens/hooked_transformer.py#L119
Unexpected token at end of expression
|