diff --git a/QEfficient/peft/auto.py b/QEfficient/peft/auto.py index bcdd79bdf..76c227862 100644 --- a/QEfficient/peft/auto.py +++ b/QEfficient/peft/auto.py @@ -39,7 +39,6 @@ class QEffAutoPeftModelForCausalLM(QEFFBaseModel): Args: :model (nn.Module): PyTorch model - :finite_adapters (bool): set True to enable finite adapter mode with QEffAutoLoraModelForCausalLM class. Please refer to QEffAutoLoraModelForCausalLM for API specification. .. code-block:: python diff --git a/docs/source/hl_api.md b/docs/source/hl_api.md index 798157be0..47dd6cde8 100644 --- a/docs/source/hl_api.md +++ b/docs/source/hl_api.md @@ -18,7 +18,8 @@ ## `QEffAutoLoraModelForCausalLM` ```{eval-rst} -.. autoclass:: QEfficient.lora.auto.QEffAutoLoraModelForCausalLM +.. autoclass:: QEfficient.peft.lora.auto.QEffAutoLoraModelForCausalLM + :member-order: bysource :members: ``` @@ -50,4 +51,4 @@ :members: :show-inheritance: :exclude-members: latency_stats_bertstyle,cloud_ai_100_exec_kv_helper -``` \ No newline at end of file +```