Hugging Face's logo

consciousAI
/
question-generation-auto-t5-v1-base-s

Text2Text Generation
Transformers
PyTorch
TensorBoard
t5
Question(s) Generation
text-generation-inference
Model card Files Metrics Community
1
Anshoo Mehra commited on
Commit
2965b96
·
1 Parent(s): a82dc16

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -27,9 +27,8 @@ from transformers import (
27
  )
28
 
29
  model_checkpoint = "anshoomehra/t5-v1_1-base-squadV2AutoQgen"
30
- device = 'cuda'
31
 
32
- model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint).to(device)
33
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
34
 
35
  ## Input with prompt
 
27
  )
28
 
29
  model_checkpoint = "anshoomehra/t5-v1_1-base-squadV2AutoQgen"
 
30
 
31
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
32
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
33
 
34
  ## Input with prompt