Updated the sample code
#58
by
frankleeeee
- opened
People often just copy and paste the example code to run, but they will encounter "torch not found" error. This PR added torch import to make it smoother to copy and run.
Thanks!
patrickvonplaten
changed pull request status to
merged