Skip to content

Commit cce01e9

Browse files
committed
fix typo
1 parent 984a772 commit cce01e9

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

diffusion_model.hpp

+3-3
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
#include "mmdit.hpp"
55
#include "unet.hpp"
66

7-
struct DiffuisionModel {
7+
struct DiffusionModel {
88
virtual void compute(int n_threads,
99
struct ggml_tensor* x,
1010
struct ggml_tensor* timesteps,
@@ -24,7 +24,7 @@ struct DiffuisionModel {
2424
virtual int64_t get_adm_in_channels() = 0;
2525
};
2626

27-
struct UNetModel : public DiffuisionModel {
27+
struct UNetModel : public DiffusionModel {
2828
UNetModelRunner unet;
2929

3030
UNetModel(ggml_backend_t backend,
@@ -72,7 +72,7 @@ struct UNetModel : public DiffuisionModel {
7272
}
7373
};
7474

75-
struct MMDiTModel : public DiffuisionModel {
75+
struct MMDiTModel : public DiffusionModel {
7676
MMDiTRunner mmdit;
7777

7878
MMDiTModel(ggml_backend_t backend,

stable-diffusion.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@ class StableDiffusionGGML {
7979

8080
std::shared_ptr<Conditioner> cond_stage_model;
8181
std::shared_ptr<FrozenCLIPVisionEmbedder> clip_vision; // for svd
82-
std::shared_ptr<DiffuisionModel> diffusion_model;
82+
std::shared_ptr<DiffusionModel> diffusion_model;
8383
std::shared_ptr<AutoEncoderKL> first_stage_model;
8484
std::shared_ptr<TinyAutoEncoder> tae_first_stage;
8585
std::shared_ptr<ControlNet> control_net;

0 commit comments

Comments
 (0)