Avoid GPU OOM for a TF Rag test (#17638)

Commit

3 years ago

Avoid GPU OOM for a TF Rag test (#17638) Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>