cocktailpeanut commited on
Commit
d7bca4c
1 Parent(s): 43f619a
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ else:
16
  device = "cpu"
17
  model_id = "stabilityai/stable-diffusion-2-1-base"
18
 
19
- to = torch.float16 if self.device == 'cuda' else torch.float32
20
 
21
  # components for the Preprocessor
22
  scheduler = DDIMScheduler.from_pretrained(model_id, subfolder="scheduler")
 
16
  device = "cpu"
17
  model_id = "stabilityai/stable-diffusion-2-1-base"
18
 
19
+ to = torch.float16 if device == 'cuda' else torch.float32
20
 
21
  # components for the Preprocessor
22
  scheduler = DDIMScheduler.from_pretrained(model_id, subfolder="scheduler")