Disable return all hidden states to reduce gpu memory use#173
Open
heya5 wants to merge 2 commits intoevolutionaryscale:mainfrom
Open
Disable return all hidden states to reduce gpu memory use#173heya5 wants to merge 2 commits intoevolutionaryscale:mainfrom
heya5 wants to merge 2 commits intoevolutionaryscale:mainfrom