Spaces:

ixarchakos
/

TryOffAnyone

Running

ixarchakos commited on Jan 17

Commit

7b2c75c

verified ·

1 Parent(s): 8d55a56

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -28,7 +28,7 @@ TITLE = """
 [[github:ixarchakos/try-off-anyone]](https://github.com/ixarchakos/try-off-anyone)
 """
-DEVICE = torch.device("cuda" if torch.cuda.is_available() else "mps")
 DTYPE = torch.bfloat16 if DEVICE == 'cuda' else torch.float32
 pipeline_tryoff = TryOffAnyone(

 [[github:ixarchakos/try-off-anyone]](https://github.com/ixarchakos/try-off-anyone)
 """
+DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 DTYPE = torch.bfloat16 if DEVICE == 'cuda' else torch.float32
 pipeline_tryoff = TryOffAnyone(