Skip to content

Support for LongLLaMA tensor format of _past_key_values#390

Open
vackosar wants to merge 1 commit intoguidance-ai:mainfrom
vackosar:patch-1
Open

Support for LongLLaMA tensor format of _past_key_values#390
vackosar wants to merge 1 commit intoguidance-ai:mainfrom
vackosar:patch-1

Conversation

@vackosar
Copy link

Generalize the line in TransformerSession that trims the cache to support LongLLaMA tensor layout that has tuple length of 6 instead of 2.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant