consciousAI
/

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

Text2Text Generation

Question(s) Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

Anshoo Mehra commited on Oct 22, 2022

Commit

2965b96

•

1 Parent(s): a82dc16

Update README.md

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -27,9 +27,8 @@ from transformers import (
 )
 model_checkpoint = "anshoomehra/t5-v1_1-base-squadV2AutoQgen"
-device = 'cuda'
-model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint).to(device)
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 ## Input with prompt

 )
 model_checkpoint = "anshoomehra/t5-v1_1-base-squadV2AutoQgen"
+model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 ## Input with prompt