keras
/

gpt2_medium_en

Divyasreepat commited on 19 days ago

Commit

3785216

•

1 Parent(s): 71486a4

Update README.md with new model card content

Files changed (1) hide show

README.md CHANGED Viewed

@@ -92,7 +92,7 @@ gpt2_lm.generate(prompt)
 Call `fit()` on a single batch.
 ```python
 features = ["The quick brown fox jumped.", "I forgot my homework."]
-gpt2_lm = keras_hub.models.GPT2CausalLM.from_preset("gpt2_base_en")
 gpt2_lm.fit(x=features, batch_size=2)
 ```
@@ -158,7 +158,7 @@ gpt2_lm.generate(prompt)
 Call `fit()` on a single batch.
 ```python
 features = ["The quick brown fox jumped.", "I forgot my homework."]
-gpt2_lm = keras_hub.models.GPT2CausalLM.from_preset("gpt2_base_en")
 gpt2_lm.fit(x=features, batch_size=2)
 ```

 Call `fit()` on a single batch.
 ```python
 features = ["The quick brown fox jumped.", "I forgot my homework."]
+gpt2_lm = keras_hub.models.GPT2CausalLM.from_preset("gpt2_medium_en")
 gpt2_lm.fit(x=features, batch_size=2)
 ```
 Call `fit()` on a single batch.
 ```python
 features = ["The quick brown fox jumped.", "I forgot my homework."]
+gpt2_lm = keras_hub.models.GPT2CausalLM.from_preset("hf://keras/gpt2_medium_en")
 gpt2_lm.fit(x=features, batch_size=2)
 ```