title

Author: xadupre

_doc/examples/plot_export_tiny_llm.py

@@ -174,4 +174,4 @@ def _forward_(*args, _f=None, **kwargs):
 # If you have any error, then look at example
 # :ref:`l-plot-tiny-llm-export-patched`.
 
-doc.plot_legend("Tiny-LLM fails", "torch.export.export", "tomato")
+doc.plot_legend("Tiny-LLM\nforward inputs\nbehind generate", "torch.export.export", "tomato")