mirror of
https://github.com/leejet/stable-diffusion.cpp.git
synced 2026-02-04 19:03:35 +00:00
Fix diffusers_style detection
This commit is contained in:
parent
1241323c4a
commit
203d0539fe
4
flux.hpp
4
flux.hpp
@ -782,8 +782,8 @@ namespace Flux {
|
|||||||
bool use_yak_mlp = false;
|
bool use_yak_mlp = false;
|
||||||
bool use_mlp_silu_act = false;
|
bool use_mlp_silu_act = false;
|
||||||
float ref_index_scale = 1.f;
|
float ref_index_scale = 1.f;
|
||||||
ChromaRadianceParams chroma_radiance_params;
|
|
||||||
bool diffusers_style = false;
|
bool diffusers_style = false;
|
||||||
|
ChromaRadianceParams chroma_radiance_params;
|
||||||
};
|
};
|
||||||
|
|
||||||
struct Flux : public GGMLBlock {
|
struct Flux : public GGMLBlock {
|
||||||
@ -1308,7 +1308,7 @@ namespace Flux {
|
|||||||
// not schnell
|
// not schnell
|
||||||
flux_params.guidance_embed = true;
|
flux_params.guidance_embed = true;
|
||||||
}
|
}
|
||||||
if (tensor_name.find("model.diffusion_model.single_blocks.0.linear1.weight.1") == std::string::npos) {
|
if (tensor_name.find("model.diffusion_model.single_blocks.0.linear1.weight.1") != std::string::npos) {
|
||||||
flux_params.diffusers_style = true;
|
flux_params.diffusers_style = true;
|
||||||
}
|
}
|
||||||
if (tensor_name.find("distilled_guidance_layer.in_proj.weight") != std::string::npos) {
|
if (tensor_name.find("distilled_guidance_layer.in_proj.weight") != std::string::npos) {
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user