Update README.md
Browse files
README.md
CHANGED
@@ -81,7 +81,7 @@ prompt = f"""<schema>{schema}</schema>
|
|
81 |
|
82 |
### PyTorch
|
83 |
```python
|
84 |
-
from transformers import
|
85 |
device = "cuda"
|
86 |
model = AutoModelForCausalLM.from_pretrained("PipableAI/pipSQL-1.3b")
|
87 |
tokenizer = AutoTokenizer.from_pretrained("PipableAI/pipSQL-1.3b")
|
|
|
81 |
|
82 |
### PyTorch
|
83 |
```python
|
84 |
+
from transformers import AutoModelForCausalLM, AutoTokenizer
|
85 |
device = "cuda"
|
86 |
model = AutoModelForCausalLM.from_pretrained("PipableAI/pipSQL-1.3b")
|
87 |
tokenizer = AutoTokenizer.from_pretrained("PipableAI/pipSQL-1.3b")
|