def get_llm()

in experiments/google/cloud/ml/applied/utils/utils.py [0:0]

• 8 lines of code
• 1 McCabe index (conditional complexity)

def get_llm():
    vertexai.init(
        project=conf.value(Config.SECTION_PROJECT, "id"),
        location=conf.value(Config.SECTION_PROJECT, "location"),
    )

    return vertexai.language_models.TextGenerationModel.from_pretrained(
        conf.value(Config.SECTION_MODELS, "llm")
    )