Update README.md
Browse files
README.md
CHANGED
|
@@ -101,7 +101,7 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True).split('<sql>')[1].s
|
|
| 101 |
|
| 102 |
### Flax
|
| 103 |
```python
|
| 104 |
-
from
|
| 105 |
device = "cuda"
|
| 106 |
model = FlaxAutoModelForCausalLM.from_pretrained("PipableAI/pipSQL-1.3b")
|
| 107 |
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
|
|
| 113 |
|
| 114 |
### TensorFlow
|
| 115 |
```python
|
| 116 |
-
from
|
| 117 |
device = "cuda"
|
| 118 |
model = TFAutoModelForCausalLM.from_pretrained("PipableAI/pipSQL-1.3b")
|
| 119 |
tokenizer = AutoTokenizer.from_pretrained("PipableAI/pipSQL-1.3b")
|
|
|
|
| 101 |
|
| 102 |
### Flax
|
| 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")
|
|
|
|
| 113 |
|
| 114 |
### TensorFlow
|
| 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")
|