You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Thank you for your excellent work here. We are implementing the Idefics 2 model into mistral.rs which is built on Candle.
Candle has a Pytorch-like API, however, it does not have some functions. In particular, we noticed that Candle is missing the torch.bucketize function. I have opened an issue on Candle but thought it would also be a good idea to check here how to implement torch.bucketize using more elementary functions or generally rewriting that part of the code to emulate the behavior.
Hello everyone,
Thank you for your excellent work here. We are implementing the Idefics 2 model into
mistral.rs
which is built on Candle.Candle has a Pytorch-like API, however, it does not have some functions. In particular, we noticed that Candle is missing the
torch.bucketize
function. I have opened an issue on Candle but thought it would also be a good idea to check here how to implementtorch.bucketize
using more elementary functions or generally rewriting that part of the code to emulate the behavior.This is the part of the code I am referencing:
Is there a way to rewrite this such that I do not need to call
torch.bucketize
? Thank you for any help, I would really appreciate it!The text was updated successfully, but these errors were encountered: