Hugging Face's logo

Spaces:
pdltiet
/
demo-gpu
Running on Zero

App Files Community
2
Ayushnangia commited on
Commit
b849cff
•
1 Parent(s): 34bc65c

unable to test torch with zero

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,8 +34,8 @@ grammar_args = TTSettings(num_beams=5, min_length=1)
34
  # Spell Check Model initialization
35
  OCRtokenizer = AutoTokenizer.from_pretrained("Bhuvana/t5-base-spellchecker", use_fast=False)
36
  OCRmodel = AutoModelForSeq2SeqLM.from_pretrained("Bhuvana/t5-base-spellchecker")
37
- zero = torch.Tensor([0]).cuda()
38
- print(zero.device)
39
 
40
 
41
  def correct_spell(inputs):
 
34
  # Spell Check Model initialization
35
  OCRtokenizer = AutoTokenizer.from_pretrained("Bhuvana/t5-base-spellchecker", use_fast=False)
36
  OCRmodel = AutoModelForSeq2SeqLM.from_pretrained("Bhuvana/t5-base-spellchecker")
37
+ # zero = torch.Tensor([0]).cuda()
38
+ # print(zero.device)
39
 
40
 
41
  def correct_spell(inputs):