Hugging Face's logo

Spaces:
ddosxd
/
test-chat-zerogpu
Sleeping

App Files Community
ddosxd commited on
Commit
65440f9
·
verified ·
1 Parent(s): d3d24b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ model = AutoModelForCausalLM.from_pretrained(
10
  torch_dtype=torch.float16,
11
  )
12
 
13
- #zero = torch.Tensor([0]).cuda()
14
  #print(zero.device) # <-- 'cpu' 🤔
15
 
16
  @spaces.GPU
 
10
  torch_dtype=torch.float16,
11
  )
12
 
13
+ zero = torch.Tensor([0]).cuda()
14
  #print(zero.device) # <-- 'cpu' 🤔
15
 
16
  @spaces.GPU