diff --git a/diffusion_model.hpp b/diffusion_model.hpp index 4c04abe..fb28494 100644 --- a/diffusion_model.hpp +++ b/diffusion_model.hpp @@ -4,7 +4,7 @@ #include "mmdit.hpp" #include "unet.hpp" -struct DiffuisionModel { +struct DiffusionModel { virtual void compute(int n_threads, struct ggml_tensor* x, struct ggml_tensor* timesteps, @@ -24,7 +24,7 @@ struct DiffuisionModel { virtual int64_t get_adm_in_channels() = 0; }; -struct UNetModel : public DiffuisionModel { +struct UNetModel : public DiffusionModel { UNetModelRunner unet; UNetModel(ggml_backend_t backend, @@ -72,7 +72,7 @@ struct UNetModel : public DiffuisionModel { } }; -struct MMDiTModel : public DiffuisionModel { +struct MMDiTModel : public DiffusionModel { MMDiTRunner mmdit; MMDiTModel(ggml_backend_t backend, diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp index 3327fac..34bf8f5 100644 --- a/stable-diffusion.cpp +++ b/stable-diffusion.cpp @@ -79,7 +79,7 @@ public: std::shared_ptr cond_stage_model; std::shared_ptr clip_vision; // for svd - std::shared_ptr diffusion_model; + std::shared_ptr diffusion_model; std::shared_ptr first_stage_model; std::shared_ptr tae_first_stage; std::shared_ptr control_net;