Update README.md

README.md

@@ -103,7 +103,7 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True).split('<sql>')[1].s
 ```python
 from transformers import FlaxAutoModelForCausalLM, AutoTokenizer
 device = "cuda"
-model = FlaxAutoModelForCausalLM.from_pretrained("PipableAI/pip-sql-1.3b")
+model = FlaxAutoModelForCausalLM.from_pretrained("PipableAI/pip-sql-1.3b",from_pt=True)
 tokenizer = AutoTokenizer.from_pretrained("PipableAI/pip-sql-1.3b")
 
 inputs = tokenizer(text, return_tensors="jax")