refactor: fix flux

stduhpf · stduhpf · commit 371e18ba14c5 · 2024-11-05T22:53:11.000+01:00
diff --git a/flux.hpp b/flux.hpp
@@ -35,7 +35,7 @@ namespace Flux {
         int64_t hidden_size;
         float eps;
 
-        void init_params(struct ggml_context* ctx, const std::string prefix, std::map<std::string, enum ggml_type>& tensor_types, std::map<std::string, struct ggml_tensor*>& params) {
+        void init_params(struct ggml_context* ctx, std::map<std::string, enum ggml_type>& tensor_types, const std::string prefix = "") {
             ggml_type wtype = GGML_TYPE_F32;  //(tensor_types.find(prefix + "scale") != tensor_types.end()) ? tensor_types[prefix + "scale"] : GGML_TYPE_F32;
             params["scale"] = ggml_new_tensor_1d(ctx, wtype, hidden_size);
         }

Original file line number	Diff line number	Diff line change
`@@ -35,7 +35,7 @@ namespace Flux {`
`35`	`35`	`int64_t hidden_size;`
`36`	`36`	`float eps;`
`37`	`37`
`38`		`- void init_params(struct ggml_context* ctx, const std::string prefix, std::map<std::string, enum ggml_type>& tensor_types, std::map<std::string, struct ggml_tensor*>& params) {`
	`38`	`+ void init_params(struct ggml_context* ctx, std::map<std::string, enum ggml_type>& tensor_types, const std::string prefix = "") {`
`39`	`39`	`ggml_type wtype = GGML_TYPE_F32; //(tensor_types.find(prefix + "scale") != tensor_types.end()) ? tensor_types[prefix + "scale"] : GGML_TYPE_F32;`
`40`	`40`	`params["scale"] = ggml_new_tensor_1d(ctx, wtype, hidden_size);`
`41`	`41`	`}`