Skip to content

Commit 9b78b62

Browse files
authored
Linear leejet#2
1 parent 48cd384 commit 9b78b62

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

ggml_extend.hpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1274,6 +1274,7 @@ class Linear : public UnaryBlock {
12741274

12751275
void init_params(struct ggml_context* ctx, std::map<std::string, enum ggml_type>& tensor_types, const std::string prefix = "") {
12761276
if (tensor_types.find(prefix + "A") != tensor_types.end()) {
1277+
enum ggml_type wtype = tensor_types[prefix + "A"];
12771278
params["A"] = ggml_new_tensor_2d(ctx, wtype, in_features, 64);
12781279
params["B"] = ggml_new_tensor_2d(ctx, wtype, 64, out_features);
12791280
} else {

0 commit comments

Comments
 (0)