Shuyi Wang
Shuyi Wang
@a710128 Thanks for the quick response. Please keep me updated.
@a710128 I tried to import all 3 models. Surprisingly, CPM1 is fine. It started downloading after `model = bminf.models.CPM1()`. However, CPM2 and EVA reported the same CUDA OOM error. Seems...
@a710128 Could you share your installation script and cuda version?
@a710128 Actually the previous logs are not matched with my latest runs. The error actually comes from the T5 model file during the first init of the pinned decoder layer....
@a710128 Thanks for the fix. I tried 1.0.2. The import works fine for me but the inference does not. Here is the latest error: ``` >>> import bminf >>> cpm2...