From 5c561eab31033b6190ea539522a66f0ba8f21a92 Mon Sep 17 00:00:00 2001 From: leejet Date: Sun, 25 Aug 2024 16:01:36 +0800 Subject: [PATCH] feat: do not convert more flux tensors --- model.cpp | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/model.cpp b/model.cpp index 6ca7e9bc..7390c161 100644 --- a/model.cpp +++ b/model.cpp @@ -1740,11 +1740,14 @@ bool ModelLoader::tensor_should_be_converted(const TensorStorage& tensor_storage // Pass, do not convert } else if (ends_with(name, ".bias")) { // Pass, do not convert + } else if (ends_with(name, ".scale")) { + // Pass, do not convert } else if (contains(name, "img_in.") || - contains(name, "time_in.in_layer.") || - contains(name, "vector_in.in_layer.") || - contains(name, "guidance_in.in_layer.") || - contains(name, "final_layer.linear.")) { + contains(name, "txt_in.") || + contains(name, "time_in.") || + contains(name, "vector_in.") || + contains(name, "guidance_in.") || + contains(name, "final_layer.")) { // Pass, do not convert. For FLUX } else if (contains(name, "x_embedder.") || contains(name, "t_embedder.") ||