From 462d800eaf3e2a85360aa59a1a469d314e0c843f 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 95a0dd8..ace87da 100644 --- a/langfun/core/llms/__init__.py +++ b/langfun/core/llms/__init__.py @@ -47,7 +47,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 437ccb9..2f28b60 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