ChatTTS icon indicating copy to clipboard operation
ChatTTS copied to clipboard

"Compile” doesn't work on GPU

Open encorechow opened this issue 7 months ago • 1 comments

pytorch version: 2.3.1+cu121

I've tried to use compile=True in load function, it is super slow to generate results, and I didn't see any GPU utilizition with nvidia-smi like below (a liitle memory ocuupied like 1GB): 企业微信截图_17194899582803

after I commented out the code block below, GPU utilization turns normal, and result is generated much much faster: 企业微信截图_17194900322602 企业微信截图_17194900872413

I cant understand why the torch.compile function didn't work well. Does anyone know why?

encorechow avatar Jun 27 '24 12:06 encorechow