feat: do not convert more flux tensors

leejet · Aug 25, 2024 · 5c561ea · 5c561ea
1 parent f5997a1
commit 5c561ea
Showing 1 changed file with 7 additions and 4 deletions.
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.") ||