search

haofanwang / ControlNet-for-Diffusers

Transfer the ControlNet with any basemodel in diffusers🔥
MIT License
813 stars 48 forks source link

'Segmentation fault (core dumped)' #32

Open blx0102 opened 1 year ago

blx0102 commented 1 year ago

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 commented 1 year ago

Could you provide your full scripts?

maobenz commented 1 year ago

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.