File tree 2 files changed +4
-4
lines changed
2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change 4
4
#include " mmdit.hpp"
5
5
#include " unet.hpp"
6
6
7
- struct DiffuisionModel {
7
+ struct DiffusionModel {
8
8
virtual void compute (int n_threads,
9
9
struct ggml_tensor * x,
10
10
struct ggml_tensor * timesteps,
@@ -24,7 +24,7 @@ struct DiffuisionModel {
24
24
virtual int64_t get_adm_in_channels () = 0;
25
25
};
26
26
27
- struct UNetModel : public DiffuisionModel {
27
+ struct UNetModel : public DiffusionModel {
28
28
UNetModelRunner unet;
29
29
30
30
UNetModel (ggml_backend_t backend,
@@ -72,7 +72,7 @@ struct UNetModel : public DiffuisionModel {
72
72
}
73
73
};
74
74
75
- struct MMDiTModel : public DiffuisionModel {
75
+ struct MMDiTModel : public DiffusionModel {
76
76
MMDiTRunner mmdit;
77
77
78
78
MMDiTModel (ggml_backend_t backend,
Original file line number Diff line number Diff line change @@ -79,7 +79,7 @@ class StableDiffusionGGML {
79
79
80
80
std::shared_ptr<Conditioner> cond_stage_model;
81
81
std::shared_ptr<FrozenCLIPVisionEmbedder> clip_vision; // for svd
82
- std::shared_ptr<DiffuisionModel > diffusion_model;
82
+ std::shared_ptr<DiffusionModel > diffusion_model;
83
83
std::shared_ptr<AutoEncoderKL> first_stage_model;
84
84
std::shared_ptr<TinyAutoEncoder> tae_first_stage;
85
85
std::shared_ptr<ControlNet> control_net;
You can’t perform that action at this time.
0 commit comments