fix schnell support

leejet · leejet · commit 77ca8e3f48a0 · 2024-08-21T21:44:32.000+08:00
diff --git a/flux.hpp b/flux.hpp
@@ -840,7 +840,9 @@ struct FluxRunner : public GGMLRunner {
         context   = to_backend(context);
         y         = to_backend(y);
         timesteps = to_backend(timesteps);
-        guidance  = to_backend(guidance);
+        if (flux_params.guidance_embed) {
+            guidance  = to_backend(guidance);
+        }
 
         pe_vec = flux.gen_pe(x->ne[1], x->ne[0], 2, x->ne[3], context->ne[1], flux_params.theta, flux_params.axes_dim);
         int pos_len = pe_vec.size() / flux_params.axes_dim_sum / 2;
