Shorten gpt-2 model output length

deliahu · deliahu · commit 1a56661bee26 · 2019-09-24T19:21:19.000-07:00
diff --git a/examples/text-generator/gpt-2.ipynb b/examples/text-generator/gpt-2.ipynb
@@ -241,7 +241,7 @@
         "        builder.save()\n",
         "\n",
         "\n",
-        "export_for_serving(top_k=40, model_name=MODEL_SIZE)"
+        "export_for_serving(top_k=40, length=256, model_name=MODEL_SIZE)"
       ],
       "execution_count": 0,
       "outputs": []
