Skip to content

Commit dec843b

Browse files
authored
fix documentation (#18)
1 parent e9720f7 commit dec843b

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

_doc/examples/plot_export_tiny_llm.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,8 @@ def _forward_(*args, _f=None, **kwargs):
8484
#
8585
# Let's create an untrained model using the config file provided
8686
# `config.json <https://huggingface.co/arnir0/Tiny-LLM/blob/main/config.json>`_
87-
# to create an untrained model: :func:`....get_tiny_llm`.
87+
# to create an untrained model:
88+
# :func:`onnx_diagnostic.torch_models.llms.get_tiny_llm`.
8889
# Then let's use it.
8990

9091
experiment = get_tiny_llm()
@@ -138,7 +139,7 @@ def _forward_(*args, _f=None, **kwargs):
138139
#
139140
# Let's use the same dummy inputs but we use the downloaded model.
140141
# Dummy inputs and dynamic shapes are created by function
141-
# :func:`....get_tiny_llm`.
142+
# :func:`onnx_diagnostic.torch_models.llms.get_tiny_llm`.
142143

143144
data = get_tiny_llm()
144145
inputs, dynamic_shapes = data["inputs"], data["dynamic_shapes"]

0 commit comments

Comments
 (0)