Skip to content

The size of tensor a must match the size of tensor at non-singleton dimension #951

Answered by jongwook
atrdev-rgb asked this question in Q&A
Discussion options

You must be logged in to vote

Hi, it appears that you're calling the model from different threads. The model is not equipped for that, mainly because of the kv cache logic using the hooks. I'd suggest keep using the lock, if that's not too much of a slowdown.

Replies: 2 comments 2 replies

Comment options

You must be logged in to vote
0 replies
Answer selected by jongwook
Comment options

You must be logged in to vote
2 replies
@Guneetsinghtuli
Comment options

@itsomar278
Comment options

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
4 participants