keras
/

gpt2_base_en_cnn_dailymail

Divyasreepat commited on 8 days ago

Commit

4f0fbd7

•

1 Parent(s): b804d9e

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_base_en_cnn_dailymail")
 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_base_en_cnn_dailymail")
 gpt2_lm.fit(x=features, batch_size=2)
 ```