feat: token merging for image classification #1255
rensortinoAnnotations
5 errors
|
linting (3.11)
The process '/opt/hostedtoolcache/ruff/0.15.5/x86_64/ruff' failed with exit code 1
|
|
ruff (F401):
src/pruna/algorithms/token_merging.py#L424
src/pruna/algorithms/token_merging.py:424:62: F401 `transformers.models.vit.modeling_vit.ViTLayer` imported but unused; consider using `importlib.util.find_spec` to test for availability
help: Remove unused import: `transformers.models.vit.modeling_vit.ViTLayer`
|
|
ruff (F401):
src/pruna/algorithms/token_merging.py#L21
src/pruna/algorithms/token_merging.py:21:26: F401 `transformers.ImageClassificationPipeline` imported but unused
help: Remove unused import: `transformers.ImageClassificationPipeline`
|
|
ruff (I001):
src/pruna/algorithms/token_merging.py#L14
src/pruna/algorithms/token_merging.py:14:1: I001 Import block is un-sorted or un-formatted
help: Organize imports
|
|
test (3.11)
Process completed with exit code 2.
|