mirror of
https://github.com/leejet/stable-diffusion.cpp.git
synced 2026-06-09 15:56:39 +00:00
fix: load TAESD preview-only model correctly (#1547)
This commit is contained in:
parent
3a8788cb7d
commit
8cf55a3b3b
@ -759,7 +759,7 @@ public:
|
|||||||
LOG_INFO("using TAE for preview");
|
LOG_INFO("using TAE for preview");
|
||||||
preview_vae = create_tae();
|
preview_vae = create_tae();
|
||||||
preview_vae->set_max_graph_vram_bytes(max_graph_vram_bytes);
|
preview_vae->set_max_graph_vram_bytes(max_graph_vram_bytes);
|
||||||
get_param_tensors_p(first_stage_model, vae_mmap, "vae");
|
get_param_tensors_p(preview_vae, vae_mmap, "tae");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user