Make compatible with recent versions of triton

#32
by tanhevg - opened

Many people are complaining about not being able to run this model with the latest versions of triton, e.g. here. This patch solves this issue.

Ready to merge
This branch is ready to get merged automatically.
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment