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.

patrickvonplaten Stability AI org


patrickvonplaten changed pull request status to merged

Sign up or log in to comment