Segmentation Fault error in inference.py in a CPU only server

zyainfal / One-Shot-Face-Swapping-on-Megapixels

One Shot Face Swapping on Megapixels.

Other

317 stars 41 forks source link

Segmentation Fault error in inference.py in a CPU only server #50

Open arnold408 opened 1 year ago

arnold408 commented 1 year ago

I am trying to the code in a server with CPU (no GPU). Getting Segmentation Fault , core dumped error in inference.py at this line : fakeswap, = self.generator(att_struct, [swapped_lats, None], randomize_noise=False)

my torch version is 1.12.1+cpu

Please suggest. Will the code work with CPU?

Thanks