Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -87,9 +87,9 @@ print(zero.device)
|
|
| 87 |
|
| 88 |
@spaces.GPU
|
| 89 |
def test():
|
| 90 |
-
print('intest', zero.device)
|
| 91 |
device = zero.device
|
| 92 |
-
print('1 ',torch.Tensor([1]).
|
| 93 |
return 0
|
| 94 |
|
| 95 |
test()
|
|
|
|
| 87 |
|
| 88 |
@spaces.GPU
|
| 89 |
def test():
|
| 90 |
+
print('intest device', zero.device)
|
| 91 |
device = zero.device
|
| 92 |
+
print('1 ',torch.Tensor([1]).cuda())
|
| 93 |
return 0
|
| 94 |
|
| 95 |
test()
|