lmxyy / sige Goto Github PK

Hi @lmxyy ,
Thank you so much for your awesome work!
It is a very interesting topic, and I tried to run the example.py.
However, It was not successful. Could you help give me some advice?

Traceback (most recent call last):
File "/xx/sige-main/example.py", line 98, in
main()
File "/xx/sige-main/example.py", line 87, in main
sige_output = model(edited_input)
File "/xx/anaconda2/envs/pt1.7/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
result = self.forward(*input, **kwargs)
File "/xx/sige-main/example.py", line 47, in forward
return self.example_module(x)
File "/xx/anaconda2/envs/pt1.7/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
result = self.forward(*input, **kwargs)
File "/xx/sige-main/example.py", line 33, in forward
x = self.gather(x)
File "/xx/anaconda2/envs/pt1.7/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
result = self.forward(*input, **kwargs)
File "/xx/sige-main/sige/nn/gather.py", line 79, in forward
assert runtime is not None
AssertionError

Thanks!

Hi @lmxyy! Thanks for the awesome work and open-sourced code base!

I was wondering if you plan to release the code for stable diffusion image editing as mentioned in the paper?

Thanks a lot in advance!