Problem reproducing the example images

[abdo-eldesokey-kaust]

Hi, Thank you for releasing the code in this nice and clean form! I am having two issues:

• The seed specified in the notebook "123456" does not produce the intended images for the monkey and the cat!
• The optimization takes around 5 minutes on an A100 unlike what is mentioned in the paper (1 minute). Best regards,