Sylvain Gugger
Sylvain Gugger
Please use the [forums](https://discuss.huggingface.co/) to discuss questions like this as we keep the issues for bugs and feature requests only.
cc @pacman100
cc @sanchit-gandhi
Thanks for flagging! The PR linked above should fix this.
cc @gante and @ArthurZucker
It's going to be hard to help you without any reproducer.
Actually, just checked the modeling file and this function is only used in this class, so it would be cleaner to just make it a method. Could you update your...
Ok so let's keep it as a function in that module. I'd still prefer a real method (that directly returns the result of the function) to setting a function attribute...
cc @younesbelkada
Note that the `parallelize` API is going to be deprecated soon. You should load your model like this to use Accelerate instead: ```python model = T5ForConditionalGeneration.from_pretrained("google/byt5-xl", device_map="balanced") ```