'Segmentation fault (core dumped)'

[blx0102]

Hi mate, when I was using tool_transfer_control.py, I got the segmentation fault error, all the models were downloaded from official link.
Do you have any idea to solve it?

[haofanwang]

Could you provide your full scripts?

[maobenz]

actually i have met the same problem. I think we can put the tensor to gpu to solve the problem. It is because the cpu resource is too tight.

p_new = p.cuda() + input_state_dict[sd15_key_name].cuda() - sd15_state_dict[sd15_key_name].cuda()

Then I can solve the problem.