Spaces:
Running
Running
cocktailpeanut
commited on
Commit
·
a5c3e20
1
Parent(s):
007c79c
update
Browse files
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)
|