Attribute Error: Can't pickle local object

[divastar]

Hi. Thank you for sharing!
I am trying to run with :
--ni --config celeba_hq.yml --doc celeba --timesteps 20 --eta 0.85 --etaB 1 --deg sr4 --sigma_0 0.05 -i celeba_hq_sr4_sigma_0.05
but I am getting:
Attribute Error: Can't pickle local object 'Diffusion.sample_sequence..seed_worker'
in the runners/diffusion.py?
is there any way around this?
thank you