Skip to content

Commit

Permalink
Fix wavenet head check
Browse files Browse the repository at this point in the history
  • Loading branch information
mikeoliphant committed Aug 1, 2024
1 parent 028e648 commit d573155
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion NAM/get_dsp.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -172,7 +172,7 @@ std::unique_ptr<DSP> get_dsp(dspData& conf)
layer_config["channels"], layer_config["kernel_size"], layer_config["dilations"],
layer_config["activation"], layer_config["gated"], layer_config["head_bias"]));
}
const bool with_head = config["head"] == NULL;
const bool with_head = !config["head"].is_null();
const float head_scale = config["head_scale"];
out = std::make_unique<wavenet::WaveNet>(layer_array_params, head_scale, with_head, weights, expectedSampleRate);
}
Expand Down

0 comments on commit d573155

Please sign in to comment.