Updated the sample code (#58)

- Updated the sample code (f8754d06e54f1806a4eb097213eaa3bf3a668e40)


Co-authored-by: Frank Lee <frankleeeee@users.noreply.huggingface.co>

README.md

@@ -43,6 +43,7 @@ pip install diffusers transformers accelerate scipy safetensors
 Running the pipeline (if you don't swap the scheduler it will run with the default DDIM, in this example we are swapping it to DPMSolverMultistepScheduler):
 
 ```python
+import torch
 from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
 
 model_id = "stabilityai/stable-diffusion-2-1"