ipex-llm/docs/readthedocs/source/doc/PythonAPI/LLM/transformers.rst
SONG Ge f4164e4492 [BigDL LLM] Update readme for unifying transformers API (#8737)
* update readme doc

* fix readthedocs error

* update comment

* update exception error info

* invalidInputError instead

* fix readme typo error and remove import error

* fix more typo
2023-08-16 14:22:32 +08:00

92 lines
2.2 KiB
ReStructuredText

BigDL-LLM Transformers API
=====================
llm.transformers.model
---------------------------
.. autoclass:: bigdl.llm.transformers.model.AutoModelForCausalLM
:members:
:undoc-members:
:show-inheritance:
.. automethod:: from_pretrained
.. automethod:: load_convert
.. automethod:: load_low_bit
.. autoclass:: bigdl.llm.transformers.model.AutoModel
:members:
:undoc-members:
:show-inheritance:
.. automethod:: from_pretrained
.. automethod:: load_convert
.. automethod:: load_low_bit
.. autoclass:: bigdl.llm.transformers.model.AutoModelForSpeechSeq2Seq
:members:
:undoc-members:
:show-inheritance:
.. automethod:: from_pretrained
.. automethod:: load_convert
.. automethod:: load_low_bit
.. autoclass:: bigdl.llm.transformers.model.AutoModelForSeq2SeqLM
:members:
:undoc-members:
:show-inheritance:
.. automethod:: from_pretrained
.. automethod:: load_convert
.. automethod:: load_low_bit
llm.transformers.modelling_bigdl
----------------------------------------
.. autoclass:: bigdl.llm.transformers.modelling_bigdl.LlamaForCausalLM
:members:
:undoc-members:
:show-inheritance:
:exclude-members: GGML_Model, GGML_Module, HF_Class
.. automethod:: from_pretrained
.. autoclass:: bigdl.llm.transformers.modelling_bigdl.ChatGLMForCausalLM
:members:
:undoc-members:
:show-inheritance:
:exclude-members: GGML_Model, GGML_Module, HF_Class
.. automethod:: from_pretrained
.. autoclass:: bigdl.llm.transformers.modelling_bigdl.GptneoxForCausalLM
:members:
:undoc-members:
:show-inheritance:
:exclude-members: GGML_Model, GGML_Module, HF_Class
.. automethod:: from_pretrained
.. autoclass:: bigdl.llm.transformers.modelling_bigdl.BloomForCausalLM
:members:
:undoc-members:
:show-inheritance:
:exclude-members: GGML_Model, GGML_Module, HF_Class
.. automethod:: from_pretrained
.. autoclass:: bigdl.llm.transformers.modelling_bigdl.StarcoderForCausalLM
:members:
:undoc-members:
:show-inheritance:
:exclude-members: GGML_Model, GGML_Module, HF_Class
.. automethod:: from_pretrained