transformers
feat(generation): add GenerationActivations utility for stacking hidden states from generate()
#46311
Closed
Closed
feat(generation): add GenerationActivations utility for stacking hidden states from generate() #46311
jacopopper
wants to merge
5 commits into
huggingface:main
from
add-generation-activations
feat(generation): add GenerationActivations utility for stacking hidd…
e4e9c9f8
fix(generation): correct GenerationActivations docstrings and mark te…
785c691b
feat(generation): add layer/ last_layer/ last_token/ mean_pool_tokens…
b9a03060
test(generation): add projection property tests for GenerationActivat…
d22b1007
fix(generation): use Any instead of ModelOutput for gen_output type hint
6166c306
Reviewers
No reviews
Assignees
No one assigned
Labels
Code agent slop
Milestone
No milestone
Login to write a write a comment.
Login via GitHub