diff --git a/model.cpp b/model.cpp index 6ca7e9b..7390c16 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.") ||