mirror of
https://github.com/LostRuins/koboldcpp.git
synced 2025-09-12 09:59:41 +00:00
sync with sd.cpp
This commit is contained in:
parent
e5af9b5ea9
commit
186227fc26
8 changed files with 234 additions and 82 deletions
|
@ -137,8 +137,9 @@ struct FluxModel : public DiffusionModel {
|
|||
FluxModel(ggml_backend_t backend,
|
||||
std::map<std::string, enum ggml_type>& tensor_types,
|
||||
SDVersion version = VERSION_FLUX,
|
||||
bool flash_attn = false)
|
||||
: flux(backend, tensor_types, "model.diffusion_model", version, flash_attn) {
|
||||
bool flash_attn = false,
|
||||
bool use_mask = false)
|
||||
: flux(backend, tensor_types, "model.diffusion_model", version, flash_attn, use_mask) {
|
||||
}
|
||||
|
||||
void alloc_params_buffer() {
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue