AquilaX-AI
/

NL-JSON-248M

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Mr-Vicky-01 commited on Jul 23

Commit

ef02e9b

•

1 Parent(s): a174a52

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -50,8 +50,8 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
 import time
-tokenizer = AutoTokenizer.from_pretrained("AquilaX-AI/NL-JSON-77M")
-model = AutoModelForSeq2SeqLM.from_pretrained("AquilaX-AI/NL-JSON-77M")
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

 import torch
 import time
+tokenizer = AutoTokenizer.from_pretrained("AquilaX-AI/NL-JSON-248M")
+model = AutoModelForSeq2SeqLM.from_pretrained("AquilaX-AI/NL-JSON-248M")
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")