feat: do not convert more flux tensors

leejet · leejet · commit 5c561eab3103 · 2024-08-25T16:01:36.000+08:00
diff --git 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.") ||
