diff --git a/QEfficient/cloud/finetune.py b/QEfficient/cloud/finetune.py index b5e3eb62..b1f7cbc2 100644 --- a/QEfficient/cloud/finetune.py +++ b/QEfficient/cloud/finetune.py @@ -45,6 +45,14 @@ def main(**kwargs): + """ + Helper function to finetune the model on QAic. + + .. code-block:: bash + + python -m QEfficient.cloud.finetune OPTIONS + + """ # update the configuration for the training process train_config = TRAIN_CONFIG() update_config(train_config, **kwargs) diff --git a/docs/source/cli_api.md b/docs/source/cli_api.md index 32f3247b..603f0141 100644 --- a/docs/source/cli_api.md +++ b/docs/source/cli_api.md @@ -23,4 +23,9 @@ Use ``bash terminal``, else if using ``ZSH terminal`` then ``device_group``shoul ```{eval-rst} .. automodule:: QEfficient.cloud.export.main +``` +# `QEfficient.cloud.finetune` +```{eval-rst} + .. automodule:: QEfficient.cloud.finetune.main + ``` \ No newline at end of file