Merge branch 'main' of https://huggingface.co/spaces/runwayml/stable-diffusion-inpainting

app.py

@@ -18,8 +18,7 @@ auth_token = os.environ.get("API_TOKEN") or True
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-inpainting", dtype=torch.float16, revision="fp16", use_auth_token=auth_token)
-pipe.to(device)
+pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-inpainting", dtype=torch.float16, revision="fp16", use_auth_token=auth_token).to(device)
 
 transform = transforms.Compose([
       transforms.ToTensor(),