system freeze when using custom images

[SCWhite]

HI, really interesting project.

I'm having some problem running the CPU docker version.
it's fine when using
python infer_custom_input.py -input sample.png -gsd 0.5 -model_id 2 -output out.json
written in the instruction.

but when I give it a different file, the whole system crush,
python infer_custom_input.py -input test.png -gsd 0.5 -model_id 2 -output out.json
(I cut the image to match the size with sample.png 704*704)

and this is what I get on docker side

(704, 704, 3)
INFO:root:POST request,
Path: /
Headers:
Host: localhost:8007
Connection: keep-alive
Accept-Encoding: gzip, deflate
Accept: */*
User-Agent: python-requests/2.25.1
Content-Length: 311




Progress (50.0%) >>>>>>>>>>>>>>>>>>>>--------------------('GPU time (pass 1):', 3.8675999641418457)
('Decode time (pass 1):', 0.06645011901855469)
Progress (100.0%) >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>('GPU time (pass 2):', 3.893293857574463)
('Decode time (pass 2):', 0.06614208221435547)
begin

it stop at "begin" forever!

And another thing, is there any specific input format for the custom file?
seems like only take 24 bit-depth png?

any suggestion will be helpful, thanks

[caiya55]

did you solve the problem?