This commit is contained in:
wailovet
2024-06-01 03:49:40 +08:00
parent 9bb6e8919d
commit 5da9def874
2 changed files with 6 additions and 8 deletions
+4 -7
View File
@@ -1,5 +1,4 @@
from ..mz_prompt_utils import Utils
NODE_CLASS_MAPPINGS = {
}
@@ -8,11 +7,12 @@ NODE_DISPLAY_NAME_MAPPINGS = {
}
import importlib
import importlib
from . import mz_llama3
from . import mz_phi3
from .. import mz_llama_cpp
from .. import mz_llama_core_nodes
from . import mz_deprecated
AUTHOR_NAME = u"MinusZone"
@@ -75,7 +75,7 @@ class MZ_LLama3CLIPTextEncode:
FUNCTION = "encode"
CATEGORY = CATEGORY_NAME
def encode(self, **kwargs):
def encode(self, **kwargs):
kwargs["llama_cpp_model"] = kwargs.get(
"llama_cpp_model", "").replace("[downloaded]", "")
@@ -176,12 +176,9 @@ class MZ_BaseLLamaCPPCLIPTextEncode:
CATEGORY = CATEGORY_NAME
def encode(self, **kwargs):
importlib.reload(mz_llama3)
kwargs["llama_cpp_model"] = kwargs.get(
"llama_cpp_model", "").replace("[downloaded]", "")
text = mz_llama_cpp.base_query_beautify_prompt_text(kwargs)
text = mz_deprecated.base_query_beautify_prompt_text(kwargs)
conditionings = None
clip = kwargs.get("clip", None)
if clip is not None:
+2 -1
View File
@@ -1,6 +1,7 @@
import json
from .. import mz_prompt_utils
from .. import mz_llama_cpp
from . import mz_deprecated
phi3_models = [
"Phi-3-mini-4k-instruct-q4.gguf"
@@ -46,7 +47,7 @@ def query_beautify_prompt_text(args_dict):
model_file = mz_prompt_utils.Utils.hf_download_model(model_url)
args_dict["llama_cpp_model"] = model_file
full_response = mz_llama_cpp.base_query_beautify_prompt_text(
full_response = mz_deprecated.base_query_beautify_prompt_text(
args_dict=args_dict)
return full_response