Imports in diffusion_policy_state_pusht_demo notebook fail in Google Colab runtime

real-stanford / diffusion_policy

[RSS 2023] Diffusion Policy Visuomotor Policy Learning via Action Diffusion

https://diffusion-policy.cs.columbia.edu/

MIT License

1.1k stars 206 forks source link

Closed alex-klimov closed 8 months ago

alex-klimov commented 8 months ago

Hello guys,

Very impressive work. Would love to try it out with state-based notebook, but fail to run in Google Colab runtime. Import cell fails with:

XlaRuntimeError: FAILED_PRECONDITION: DNN library initialization failed. Look at the errors above for more details.

on this line:

from diffusers.training_utils import EMAModel

Could you please advise on fix/workaround?

cheng-chi commented 8 months ago

Hi @alex-klimov I have updated the notebooks! Fixed by updating diffusers to 0.18.2

alex-klimov commented 8 months ago

Thank you! I will check it out soon. I got the project to work on my local machine, so not blocked on my side.