PipableAI
/

pip-SQL-1B

Text Generation

instruction_tuned

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

QagentS commited on Feb 3

Commit

6ccc775

•

1 Parent(s): 3f432b5

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -42,8 +42,8 @@ text = """<schema>{schema}</schema>
 ```python
 from transformers import AutoModelForCasualLM, AutoTokenizer
 device = "cuda"
-model = AutoModelForCausalLM.from_pretrained("PipableAI/pipSQL")
-tokenizer = AutoTokenizer.from_pretrained("PipableAI/pipSQL")
 inputs = tokenizer(text, return_tensors="pt")
 outputs = model.generate(**inputs, max_new_tokens=200)

 ```python
 from transformers import AutoModelForCasualLM, AutoTokenizer
 device = "cuda"
+model = AutoModelForCausalLM.from_pretrained("PipableAI/pipSQL1b")
+tokenizer = AutoTokenizer.from_pretrained("PipableAI/pipSQL1b")
 inputs = tokenizer(text, return_tensors="pt")
 outputs = model.generate(**inputs, max_new_tokens=200)