to fix not running error

#20
No description provided.

change the version off torch to fix it

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment