Spaces:

LTT
/

PRM

Running on Zero

LTT commited on 27 days ago

Commit

d0b104b

•

1 Parent(s): 9330b46

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ from huggingface_hub import hf_hub_download
 if torch.cuda.is_available() and torch.cuda.device_count() >= 2:
     device0 = torch.device('cuda:0')
-    device1 = torch.device('cuda:0')
 else:
     device0 = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     device1 = device0

 if torch.cuda.is_available() and torch.cuda.device_count() >= 2:
     device0 = torch.device('cuda:0')
+    device1 = torch.device('cuda:1')
 else:
     device0 = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     device1 = device0