Update flux.hpp

idostyle · web-flow · commit 5f441abdd393 · 2025-05-14T09:23:33.000+02:00
diff --git a/flux.hpp b/flux.hpp
@@ -864,7 +864,7 @@ namespace Flux {
                 std::string tensor_name = pair.first;
                 if (tensor_name.find("model.diffusion_model.") == std::string::npos)
                     continue;
-                if (tensor_name.find("guidance_in.in_layer.weight") != std::string::npos) {
+                if (tensor_name.find("guidance_in.in_layer.weight") != std::string::npos || tensor_name.find("guidance_in.in_layer.A") != std::string::npos) {
                     // not schnell
                     flux_params.guidance_embed = true;
                 }

Original file line number	Diff line number	Diff line change
`@@ -864,7 +864,7 @@ namespace Flux {`
`864`	`864`	`std::string tensor_name = pair.first;`
`865`	`865`	`if (tensor_name.find("model.diffusion_model.") == std::string::npos)`
`866`	`866`	`continue;`
`867`		`- if (tensor_name.find("guidance_in.in_layer.weight") != std::string::npos) {`
	`867`	`+ if (tensor_name.find("guidance_in.in_layer.weight") != std::string::npos \|\| tensor_name.find("guidance_in.in_layer.A") != std::string::npos) {`
`868`	`868`	`// not schnell`
`869`	`869`	`flux_params.guidance_embed = true;`
`870`	`870`	`}`