Update README.md with new model card content
Browse files
README.md
CHANGED
|
@@ -92,7 +92,7 @@ gpt2_lm.generate(prompt)
|
|
| 92 |
Call `fit()` on a single batch.
|
| 93 |
```python
|
| 94 |
features = ["The quick brown fox jumped.", "I forgot my homework."]
|
| 95 |
-
gpt2_lm = keras_hub.models.GPT2CausalLM.from_preset("
|
| 96 |
gpt2_lm.fit(x=features, batch_size=2)
|
| 97 |
```
|
| 98 |
|
|
@@ -158,7 +158,7 @@ gpt2_lm.generate(prompt)
|
|
| 158 |
Call `fit()` on a single batch.
|
| 159 |
```python
|
| 160 |
features = ["The quick brown fox jumped.", "I forgot my homework."]
|
| 161 |
-
gpt2_lm = keras_hub.models.GPT2CausalLM.from_preset("
|
| 162 |
gpt2_lm.fit(x=features, batch_size=2)
|
| 163 |
```
|
| 164 |
|
|
|
|
| 92 |
Call `fit()` on a single batch.
|
| 93 |
```python
|
| 94 |
features = ["The quick brown fox jumped.", "I forgot my homework."]
|
| 95 |
+
gpt2_lm = keras_hub.models.GPT2CausalLM.from_preset("gpt2_extra_large_en")
|
| 96 |
gpt2_lm.fit(x=features, batch_size=2)
|
| 97 |
```
|
| 98 |
|
|
|
|
| 158 |
Call `fit()` on a single batch.
|
| 159 |
```python
|
| 160 |
features = ["The quick brown fox jumped.", "I forgot my homework."]
|
| 161 |
+
gpt2_lm = keras_hub.models.GPT2CausalLM.from_preset("hf://keras/gpt2_extra_large_en")
|
| 162 |
gpt2_lm.fit(x=features, batch_size=2)
|
| 163 |
```
|
| 164 |
|