PipableAI
/

pip-SQL-1B

Text Generation

instruction_tuned

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

QagentS commited on Feb 14

Commit

8b92ee7

•

1 Parent(s): d4ee90d

Update README.md

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -62,8 +62,7 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True).split('<sql>')[1].s
 flax
 ```python
-from transformers import AutoModelForCasualLM, AutoTokenizer
-device = "cuda"
 model = FlaxAutoModelForCausalLM.from_pretrained("PipableAI/pipSQL1b" , from_pt=True)
 tokenizer = AutoTokenizer.from_pretrained("PipableAI/pipSQL1b")
 ```

 flax
 ```python
+from transformers import FlaxAutoModelForCasualLM, AutoTokenizer
 model = FlaxAutoModelForCausalLM.from_pretrained("PipableAI/pipSQL1b" , from_pt=True)
 tokenizer = AutoTokenizer.from_pretrained("PipableAI/pipSQL1b")
 ```