liuganghuggingface commited on
Commit
77eb38b
·
verified ·
1 Parent(s): 3204b7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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]).to(device))
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()