FinnBehrendt/patched-Diffusion-Models-UAD

TypeError: cannot pickle 'weakref' object

HaleAkrami opened this issue · 1 comments

Hello,

I am trying to run your code but I get this error:
TypeError: cannot pickle 'weakref' object

Can you please help me with it?

Thanks

Hi,
I encountered this error when I was on workstations with multiple GPUs. My code is not tested for multi-gpu settings. Either you speechify CUDA_VISIBLE_DEVICES=0 (or any gpu you'd want to use) or you change the accelerator strategy of the lightning trainer.