cocktailpeanut commited on
Commit
a5c3e20
1 Parent(s): 007c79c
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -36,6 +36,7 @@ elif torch.backends.mps.is_available():
36
  dtype = torch.float32
37
  else:
38
  device = "cpu"
 
39
  #if torch.cuda.is_available():
40
  if device != "cpu":
41
  prior_pipeline = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=dtype)#.to(device)
 
36
  dtype = torch.float32
37
  else:
38
  device = "cpu"
39
+ print(f"device={device}")
40
  #if torch.cuda.is_available():
41
  if device != "cpu":
42
  prior_pipeline = StableCascadePriorPipeline.from_pretrained("stabilityai/stable-cascade-prior", torch_dtype=dtype)#.to(device)