search

BrasD99 / HeyGenClone

A simple and open-source analogue of the HeyGen system
862 stars 172 forks source link

TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first. #3

Closed stakeswky closed 9 months ago

stakeswky commented 9 months ago

Hi, I ran the program successfully according to the tutorial, but it has the following problem: Traceback (most recent call last): File "/root/autodl-tmp/HeyGenClone/translate.py", line 44, in translate( File "/root/autodl-tmp/HeyGenClone/translate.py", line 34, in translate engine(video_filename, output_filename) File "/root/autodl-tmp/HeyGenClone/core/engine.py", line 50, in call dereverb_out = self.dereverb.split(original_audio_file) File "/root/autodl-tmp/HeyGenClone/core/dereverb.py", line 225, in split return self.pred.prediction(input) File "/root/autodl-tmp/HeyGenClone/core/dereverb.py", line 194, in prediction sources = self.demix(mix.T) File "/root/autodl-tmp/HeyGenClone/core/dereverb.py", line 122, in demix sources = self.demix_base(segmented_mix, margin_size=margin) File "/root/autodl-tmp/HeyGenClone/core/dereverb.py", line 156, in demix_base input_data_1 = -spek.cuda().numpy() if self.device.type == 'cuda' else -spek.cpu().numpy() TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

My environmental information is below: Ubuntu 20.04 python 3.9 nvidia 3080

BrasD99 commented 9 months ago

Ill check this out. Thanks for your issue.

BrasD99 commented 9 months ago

Fixed in commit