EvelynFan / FaceFormer

[CVPR 2022] FaceFormer: Speech-Driven 3D Facial Animation with Transformers
MIT License
761 stars 135 forks source link

demo.py #75

Closed Sylvia-Yuju closed 1 year ago

Sylvia-Yuju commented 1 year ago

When running the demo.py code, the following problem occurs, How can I solve 1521680011628_ pic_hd

Aiedails commented 1 year ago

Me too. So I'm simply let tgt_mask = tgt_mask.unsqueeze(0)and it works.

Sylvia-Yuju commented 1 year ago

Me too. So I'm simply let tgt_mask = tgt_mask.unsqueeze(0)and it works.

Yes, I solved it like this too