unet: refactor the refactoring

Author: stduhpf

diffusion_model.hpp

@@ -34,8 +34,7 @@ struct UNetModel : public DiffusionModel {
               std::map<std::string, enum ggml_type>& tensor_types,
               SDVersion version = VERSION_SD1,
               bool flash_attn   = false)
-        : unet(backend, version, flash_attn) {
-        unet.init_params(tensor_types, "model.diffusion_model");
+        : unet(backend, tensor_types, "model.diffusion_model", version, flash_attn) {
     }
 
     void alloc_params_buffer() {

unet.hpp

@@ -532,12 +532,11 @@ struct UNetModelRunner : public GGMLRunner {
     UnetModelBlock unet;
 
     UNetModelRunner(ggml_backend_t backend,
+                    std::map<std::string, enum ggml_type>& tensor_types,
+                    const std::string prefix,
                     SDVersion version = VERSION_SD1,
                     bool flash_attn   = false)
         : GGMLRunner(backend), unet(version, flash_attn) {
-    }
-
-    void init_params(std::map<std::string, enum ggml_type>& tensor_types, const std::string prefix) {
         unet.init(params_ctx, tensor_types, prefix);
     }