QagentS commited on
Commit
aaf6904
1 Parent(s): 002c693

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 transfomers 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,7 +113,7 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True).split('<sql>')[1].s
113
 
114
  ### TensorFlow
115
  ```python
116
- from transfomers import TFAutoModelForCausalLM, AutoTokenizer
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")