PipableAI
/

pip-sql-1.3b

Text Generation

instruction_tuned

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

QagentS commited on Feb 26

Commit

f0b9ae3

•

1 Parent(s): 89ce696

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -115,7 +115,7 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True).split('<sql>')[1].s
 ```python
 from transformers import TFAutoModelForCausalLM, AutoTokenizer
 device = "cuda"
-model = TFAutoModelForCausalLM.from_pretrained("PipableAI/pip-sql-1.3b")
 tokenizer = AutoTokenizer.from_pretrained("PipableAI/pip-sql-1.3b")
 inputs = tokenizer(text, return_tensors="tf")

 ```python
 from transformers import TFAutoModelForCausalLM, AutoTokenizer
 device = "cuda"
+model = TFAutoModelForCausalLM.from_pretrained("PipableAI/pip-sql-1.3b",from_pt=True)
 tokenizer = AutoTokenizer.from_pretrained("PipableAI/pip-sql-1.3b")
 inputs = tokenizer(text, return_tensors="tf")