Update README.md
Browse files
README.md
CHANGED
|
@@ -111,18 +111,6 @@ outputs = model.generate(**inputs, max_new_tokens=200)
|
|
| 111 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True).split('<sql>')[1].split('</sql>')[0])
|
| 112 |
```
|
| 113 |
|
| 114 |
-
### TensorFlow
|
| 115 |
-
```python
|
| 116 |
-
from transformers import TFAutoModelForCausalLM, AutoTokenizer
|
| 117 |
-
device = "cuda"
|
| 118 |
-
model = TFAutoModelForCausalLM.from_pretrained("PipableAI/pip-sql-1.3b",from_pt=True)
|
| 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)
|
| 123 |
-
print(tokenizer.decode(outputs[0], skip_special_tokens=True).split('<sql>')[1].split('</sql>')[0])
|
| 124 |
-
```
|
| 125 |
-
|
| 126 |
## Examples
|
| 127 |
|
| 128 |
### Schema
|
|
|
|
| 111 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True).split('<sql>')[1].split('</sql>')[0])
|
| 112 |
```
|
| 113 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 114 |
## Examples
|
| 115 |
|
| 116 |
### Schema
|