search

EricGuo5513 / momask-codes

Official implementation of "MoMask: Generative Masked Modeling of 3D Human Motions (CVPR2024)"
https://ericguo5513.github.io/momask/
MIT License
856 stars 73 forks source link

"baddbmm__mkl" not implemented for 'Half' #45

Closed AndresOM1102 closed 3 weeks ago

AndresOM1102 commented 6 months ago

Hello! I was trying to run the first line in the demo, but an error occured. "python gen_t2m.py --gpu_id 1 --ext exp1 --text_prompt "A person is running on a treadmill.". It is important to mention, that this is being executed without cuda or GPU, just the CPU. In the attempt of doing so, it loads CLIP correctly however in the step after it fails. The error is the following:

anaconda3/envs/momask/lib/python3.7/site-packages/torch/nn/functional.py", line 4294, in multi_head_attention_forward attn_output_weights = torch.bmm(q, k.transpose(1, 2)) RuntimeError: "baddbmm__mkl" not implemented for 'Half'

Of what I have seen, it is an error that can't be solved unless I use cuda or I change some parameters, which I don't know where to change.

Murrol commented 6 months ago

Hi,

Pleased refer to this: https://github.com/EricGuo5513/momask-codes/blob/c7825522ccf259b232aee71418f62724e84e06c3/models/mask_transformer/transformer.py#L184

AndresOM1102 commented 6 months ago

Hi,

I commented the line, however the same error appears. Do you know what else it might be?

Murrol commented 6 months ago

https://github.com/EricGuo5513/momask-codes/blob/c7825522ccf259b232aee71418f62724e84e06c3/models/mask_transformer/transformer.py#L734

AndresOM1102 commented 6 months ago

Thanks, it worked!