transformers.js
Add support for visualizing self-attention heatmaps + sequence classifier outputs w/ attentions
#1117
Merged

Add support for visualizing self-attention heatmaps + sequence classifier outputs w/ attentions #1117

xenova merged 14 commits into main from support-classifier-attentions
xenova
xenova Add support for nearest neighbour interpolation
65dd00a3
xenova Support bigint inputs for `min` and `max` helper functions
03205cfd
xenova Add min/max/argmin/argmax tensor ops
f991d895
xenova Allow sequence classifier outputs with attentions
effca0ea
HuggingFaceDocBuilderDev
xenova Align depth estimation pipeline w/ python version
66e69901
xenova Update `min` and `max` types
18596f11
xenova Fix qwen2vl processor
fca6396e
xenova Update depth estimation pipeline unit tests
10fc2ae5
xenova Remove old comments
158c7077
xenova Update types
ea5b789c
xenova Merge branch 'main' into support-classifier-attentions
c99d08ef
xenova Fix type issues
36ef7c76
xenova Merge branch 'main' into support-classifier-attentions
ac2493a4
xenova Merge branch 'main' into support-classifier-attentions
84944145
xenova xenova merged 41a6139e into main 1 year ago
xenova xenova deleted the support-classifier-attentions branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone