get_llm() function | Teradata Package for Generative AI - get_llm() - Teradata VantageCloud Lake

The get_llm() function gets the name of the large language model in use.

Example setup: Import the required modules

>>> from teradatagenai import TeradataAI

Set up the environment to work with the 'xlm-roberta-base-language-detection' Hugging Face model

model_name = 'papluca/xlm-roberta-base-language-detection'
model_args = {'transformer_class': 'AutoModelForSequenceClassification',
              'task': 'text-classification'}

llm = TeradataAI(api_type = "hugging_face",
                      model_name = model_name,
                      model_args = model_args)

Get the Hugging Face model name

>>> llm.get_llm()

Output:

'bhadresh-savani/distilbert-base-uncased-emotion'