Synthesize - MelGan: Run out of Memory with CUDA

[babysor]

Env: （Nvidia T4，torch 1.9.0）
Tried the quick start with steps:
$ conda create -n ParallelTTS python=3.7.9
$ conda activate ParallelTTS
$ pip install -r requirements.txt
$ python synthesize.py
--checkpoint ./pretrained/ljspeech-parallel-epoch0100.pth
--melgan_checkpoint ./pretrained/ljspeech-melgan-epoch3200.pth
--input_texts ./samples/english/synthesize.txt
Failed with:

[babysor]

Tried with:
--device CPU
It works.