PipableAI
/

pip-sql-1.3b

Text Generation

instruction_tuned

text-generation-inference

Model card Files Files and versions

rvk7895 commited on Feb 15, 2024

Commit

6efc277

·

verified ·

1 Parent(s): a537d08

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -81,7 +81,7 @@ prompt = f"""<schema>{schema}</schema>
 ### PyTorch
 ```python
-from transformers import AutoModelForCasualLM, AutoTokenizer
 device = "cuda"
 model = AutoModelForCausalLM.from_pretrained("PipableAI/pipSQL-1.3b")
 tokenizer = AutoTokenizer.from_pretrained("PipableAI/pipSQL-1.3b")

 ### PyTorch
 ```python
+from transformers import AutoModelForCausalLM, AutoTokenizer
 device = "cuda"
 model = AutoModelForCausalLM.from_pretrained("PipableAI/pipSQL-1.3b")
 tokenizer = AutoTokenizer.from_pretrained("PipableAI/pipSQL-1.3b")