Hugging Face's logo

PipableAI
/
pip-sql-1.3b

Text Generation
Transformers
PyTorch
Safetensors
JAX
English
llama
sql
code
text2sql
instruction_tuned
basemodel
text-generation-inference
conversational
Inference Endpoints
Model card Files Community
5
QagentS commited on
Commit
1242ee2
·
verified ·
1 Parent(s): e83b789

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -6
README.md CHANGED
@@ -91,8 +91,8 @@ prompt = f"""<schema>{schema}</schema>
91
  ```python
92
  from transformers import AutoModelForCausalLM, AutoTokenizer
93
  device = "cuda"
94
- model = AutoModelForCausalLM.from_pretrained("PipableAI/pip-sql-1.3b")
95
- tokenizer = AutoTokenizer.from_pretrained("PPipableAI/pip-sql-1.3b")
96
 
97
  inputs = tokenizer(text, return_tensors="pt")
98
  outputs = model.generate(**inputs, max_new_tokens=200)
@@ -103,8 +103,8 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True).split('<sql>')[1].s
103
  ```python
104
  from transformers import FlaxAutoModelForCausalLM, AutoTokenizer
105
  device = "cuda"
106
- model = FlaxAutoModelForCausalLM.from_pretrained("PipableAI/pip-sql-1.3b")
107
- tokenizer = AutoTokenizer.from_pretrained("PipableAI/pip-sql-1.3b")
108
 
109
  inputs = tokenizer(text, return_tensors="jax")
110
  outputs = model.generate(**inputs, max_new_tokens=200)
@@ -115,8 +115,8 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True).split('<sql>')[1].s
115
  ```python
116
  from transformers import TFAutoModelForCausalLM, AutoTokenizer
117
  device = "cuda"
118
- model = TFAutoModelForCausalLM.from_pretrained("PipableAI/pip-sql-1.3b")
119
- tokenizer = AutoTokenizer.from_pretrained("PipableAI/pip-sql-1.3b")
120
 
121
  inputs = tokenizer(text, return_tensors="tf")
122
  outputs = model.generate(**inputs, max_new_tokens=200)
 
91
  ```python
92
  from transformers import AutoModelForCausalLM, AutoTokenizer
93
  device = "cuda"
94
+ model = AutoModelForCausalLM.from_pretrained("PipableAI/pipSQL-1.3b")
95
+ tokenizer = AutoTokenizer.from_pretrained("PipableAI/pipSQL-1.3b")
96
 
97
  inputs = tokenizer(text, return_tensors="pt")
98
  outputs = model.generate(**inputs, max_new_tokens=200)
 
103
  ```python
104
  from transformers import FlaxAutoModelForCausalLM, AutoTokenizer
105
  device = "cuda"
106
+ model = FlaxAutoModelForCausalLM.from_pretrained("PipableAI/pipSQL-1.3b")
107
+ tokenizer = AutoTokenizer.from_pretrained("PipableAI/pipSQL-1.3b")
108
 
109
  inputs = tokenizer(text, return_tensors="jax")
110
  outputs = model.generate(**inputs, max_new_tokens=200)
 
115
  ```python
116
  from transformers import TFAutoModelForCausalLM, AutoTokenizer
117
  device = "cuda"
118
+ model = TFAutoModelForCausalLM.from_pretrained("PipableAI/pipSQL-1.3b")
119
+ tokenizer = AutoTokenizer.from_pretrained("PipableAI/pipSQL-1.3b")
120
 
121
  inputs = tokenizer(text, return_tensors="tf")
122
  outputs = model.generate(**inputs, max_new_tokens=200)