Update README.md code to import `torch`
#43
by
simon-mo
- opened
The sample code snippet doesn't run by itself because torch would be undefined in
StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)