alphatozeta commited on
Commit
0d93cbc
1 Parent(s): 820cf05

Changed dtype to torch_dtype

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,7 +18,8 @@ auth_token = os.environ.get("API_TOKEN") or True
18
 
19
  device = "cuda" if torch.cuda.is_available() else "cpu"
20
 
21
- pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-inpainting", dtype=torch.float16, revision="fp16", use_auth_token=auth_token).to(device)
 
22
 
23
  transform = transforms.Compose([
24
  transforms.ToTensor(),
18
 
19
  device = "cuda" if torch.cuda.is_available() else "cpu"
20
 
21
+ pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-inpainting", torch_
22
+ dtype=torch.float16, revision="fp16", use_auth_token=auth_token).to(device)
23
 
24
  transform = transforms.Compose([
25
  transforms.ToTensor(),