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 25

Commit

aaf6904

•

1 Parent(s): 002c693

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -101,7 +101,7 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True).split('<sql>')[1].s
 ### Flax
 ```python
-from transfomers import FlaxAutoModelForCausalLM, AutoTokenizer
 device = "cuda"
 model = FlaxAutoModelForCausalLM.from_pretrained("PipableAI/pipSQL-1.3b")
 tokenizer = AutoTokenizer.from_pretrained("PipableAI/pipSQL-1.3b")
@@ -113,7 +113,7 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True).split('<sql>')[1].s
 ### TensorFlow
 ```python
-from transfomers import TFAutoModelForCausalLM, AutoTokenizer
 device = "cuda"
 model = TFAutoModelForCausalLM.from_pretrained("PipableAI/pipSQL-1.3b")
 tokenizer = AutoTokenizer.from_pretrained("PipableAI/pipSQL-1.3b")

 ### Flax
 ```python
+from transformers import FlaxAutoModelForCausalLM, AutoTokenizer
 device = "cuda"
 model = FlaxAutoModelForCausalLM.from_pretrained("PipableAI/pipSQL-1.3b")
 tokenizer = AutoTokenizer.from_pretrained("PipableAI/pipSQL-1.3b")
 ### TensorFlow
 ```python
+from transformers import TFAutoModelForCausalLM, AutoTokenizer
 device = "cuda"
 model = TFAutoModelForCausalLM.from_pretrained("PipableAI/pipSQL-1.3b")
 tokenizer = AutoTokenizer.from_pretrained("PipableAI/pipSQL-1.3b")