upstream-archive/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-08-03 00:30:41 -05:00
Code Issues Packages Projects Releases Wiki Activity

refactor: split model loader from model definitions (#1619)

Browse Source
This commit is contained in:
leejet
2026-06-07 23:20:12 +08:00
committed by GitHub
parent 2a07540c2a
commit b3d56d0ba1
25 changed files with 106 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/model/diffusion/control.hpp
View File

@@ -1,8 +1,8 @@
#ifndef __SD_MODEL_DIFFUSION_CONTROL_HPP__
#define __SD_MODEL_DIFFUSION_CONTROL_HPP__
#include "model.h"
#include "model/common/block.hpp"
#include "model_loader.h"
#define CONTROL_NET_GRAPH_SIZE 1536

2
src/model/diffusion/flux.hpp
View File

@@ -4,10 +4,10 @@
#include <memory>
#include <vector>
#include "model.h"
#include "model/common/rope.hpp"
#include "model/diffusion/dit.hpp"
#include "model/diffusion/model.hpp"
#include "model_loader.h"
#define FLUX_GRAPH_SIZE 10240

1
src/model/diffusion/ltxv.hpp
View File

@@ -13,6 +13,7 @@
#include "model/common/rope.hpp"
#include "model/diffusion/flux.hpp"
#include "model/diffusion/model.hpp"
#include "model_loader.h"
namespace LTXV {

2
src/model/diffusion/mmdit.hpp
View File

@@ -7,9 +7,9 @@
#include <vector>
#include "core/ggml_extend.hpp"
#include "model.h"
#include "model/common/block.hpp"
#include "model/diffusion/model.hpp"
#include "model_loader.h"
#define MMDIT_GRAPH_SIZE 10240

1
src/model/diffusion/qwen_image.hpp
View File

@@ -6,6 +6,7 @@
#include "model/common/block.hpp"
#include "model/diffusion/flux.hpp"
#include "model/diffusion/model.hpp"
#include "model_loader.h"
namespace Qwen {
constexpr int QWEN_IMAGE_GRAPH_SIZE = 20480;

1
src/model/diffusion/wan.hpp
View File

@@ -9,6 +9,7 @@
#include "model/common/rope.hpp"
#include "model/diffusion/flux.hpp"
#include "model/diffusion/model.hpp"
#include "model_loader.h"
namespace WAN {

1
src/model/diffusion/z_image.hpp
View File

@@ -7,6 +7,7 @@
#include "model/diffusion/flux.hpp"
#include "model/diffusion/mmdit.hpp"
#include "model/diffusion/model.hpp"
#include "model_loader.h"
// Ref: https://github.com/Alpha-VLLM/Lumina-Image-2.0/blob/main/model/model.py
// Ref: https://github.com/huggingface/diffusers/pull/12703