Hi,

Importing the torch library is necessary to avoid a torch import error.

Thank you.

Thanks!

osanseviero changed pull request status to merged

Sign up or log in to comment