From 116bfa3d68071fed8b7446afe39947fc949ad55a Mon Sep 17 00:00:00 2001 From: D-X-Y Date: Sat, 7 Oct 2023 19:56:16 +0000 Subject: [PATCH] Rename as LlamaCppRemote --- langfun/core/llms/__init__.py | 2 +- langfun/core/llms/llama_cpp.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/langfun/core/llms/__init__.py b/langfun/core/llms/__init__.py index 01dea8e3..9bf3f731 100644 --- a/langfun/core/llms/__init__.py +++ b/langfun/core/llms/__init__.py @@ -46,7 +46,7 @@ from langfun.core.llms.openai import Gpt3Ada # LLaMA C++ models. -from langfun.core.llms.llama_cpp import RemoteLLaMACPP +from langfun.core.llms.llama_cpp import LlamaCppRemote # Placeholder for Google-internal imports. diff --git a/langfun/core/llms/llama_cpp.py b/langfun/core/llms/llama_cpp.py index 437ccb92..2f28b602 100644 --- a/langfun/core/llms/llama_cpp.py +++ b/langfun/core/llms/llama_cpp.py @@ -20,7 +20,7 @@ import langfun.core as lf @lf.use_init_args(['url']) -class RemoteLLaMACPP(lf.LanguageModel): +class LlamaCppRemote(lf.LanguageModel): """The remote LLaMA C++ model. The Remote LLaMA C++ models can be launched via