If the initial problem is solved, we can close this discussion. 复现方法 | steps to reproduce. I'm wondering if it's caused by pipe.enable_model_cpu_offload()?
and the output of the model is really a mess. If you are struggling with. Sign up for a free github account to open an issue and contact its maintainers and the community.
This way, you load the model only once, speeding up the process. Same model and same machine, sometimes it takes less than 1 minute, but sometimes it takes more than 10 minutes. Is there any way that checkpoint shards can maybe be cached or. Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.
Have a question about this project? 33%| | 1/3 [00:16<00:32, 16.31s/it]killedubuntu 内存不够,杀掉了pytorch占用太多内… Load the model once in a jupyter notebook cell, and run the generation in a separate cell. It said that some weights of the model checkpoint at checkpoints were not used when initializing t5forconditionalgeneration:
When working with large models in pytorch lightning,. Loading checkpoint shards is very slow. When you opened the issue it was about a timing issue, not a not able to load checkpoints issue. Saved searches use saved searches to filter your results more quickly