finalize the small model.

README.md

@@ -10,4 +10,10 @@ pinned: false
 license: mit
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference
+
+## Build wavenet-cpp
+
+
+    ./bazelisk-linux-amd64 build wavegru_mod -c opt --copt=-march=native
+    cp -f bazel-bin/wavegru_mod.so .
+

inference.py

@@ -69,7 +69,7 @@ def mel_to_wav(net, netcpp, mel, config):
     mel = np.pad(mel, [(0, 0), (pad, pad), (0, 0)], mode="edge")
     ft = wavegru_inference(net, mel)
     ft = jax.device_get(ft[0])
-    wav = netcpp.inference(ft, 1.0)
+    wav = netcpp.inference(ft, 0.9)
     wav = np.array(wav)
     wav = librosa.mu_expand(wav - 127, mu=255)
     wav = librosa.effects.deemphasis(wav, coef=0.86)

wavegru.ckpt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f1232c7c32ac43f080f18c85ad50f1313aa5b34aebd3198e56288336b012585b
+oid sha256:edd8a1ccc0a74a0b63fa416699fc0991e798d1444683be4eaf6a65249c56f8de
 size 58039876

wavegru_mod.cc

@@ -122,18 +122,18 @@ struct WaveGRU {
       }
       o1.SpMM_bias(h, o1b, &fco1, true);
       o2.SpMM_bias(fco1, o2b, &fco2, false);
-      auto max_logit = fco2[0];
-      for (int i = 1; i <= 255; ++i) {
-        max_logit = max(max_logit, fco2[i]);
-      }
-      float total = 0.0;
-      for (int i = 0; i <= 255; ++i) {
-        logits[i] = csrblocksparse::fast_exp(fco2[i] - max_logit);
-        total += logits[i];
-      }
-      for (int i = 0; i <= 255; ++i) {
-        if (logits[i] < total / 1024.0) fco2[i] = -1e9;
-      }
+      // auto max_logit = fco2[0];
+      // for (int i = 1; i <= 255; ++i) {
+      //   max_logit = max(max_logit, fco2[i]);
+      // }
+      // float total = 0.0;
+      // for (int i = 0; i <= 255; ++i) {
+      //   logits[i] = csrblocksparse::fast_exp(fco2[i] - max_logit);
+      //   total += logits[i];
+      // }
+      // for (int i = 0; i <= 255; ++i) {
+      //   if (logits[i] < total / 1024.0) fco2[i] = -1e9;
+      // }
       value = fco2.Sample(temperature);
       signal[index] = value;
     }

wavegru_mod.so

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ca652aeaa19a5e02b0c1d2606a00520176b8d896b4b29b2d368e6fe8485a7e8d
+oid sha256:700f2cade76db615b1e38bddfc9c604ff1c8ea1af3e507f879d0ceebae5d232d
 size 525536