feat: wasm would work

.gitignore

@@ -1,10 +1,10 @@
-target
+target/
 models/
 !models/.gitkeep
 venv/
 .env
-output.wav
+*.wav
-node_modules
+node_modules/
 dist/
 *.csv
 *.bin

crates/sbv2_api/Cargo.toml

@@ -6,6 +6,7 @@ description.workspace = true
 readme.workspace = true
 repository.workspace = true
 documentation.workspace = true
+license.workspace = true
 
 [dependencies]
 anyhow.workspace = true

crates/sbv2_bindings/Cargo.toml

@@ -6,6 +6,7 @@ description.workspace = true
 readme.workspace = true
 repository.workspace = true
 documentation.workspace = true
+license.workspace = true
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 [lib]

crates/sbv2_core/Cargo.toml

@@ -6,6 +6,7 @@ description.workspace = true
 readme.workspace = true
 repository.workspace = true
 documentation.workspace = true
+license.workspace = true
 
 [dependencies]
 anyhow.workspace = true

crates/sbv2_wasm/Cargo.toml

@@ -6,6 +6,7 @@ description.workspace = true
 readme.workspace = true
 repository.workspace = true
 documentation.workspace = true
+license.workspace = true
 
 [lib]
 crate-type = ["cdylib", "rlib"]

crates/sbv2_wasm/build.sh

@@ -1,5 +1,8 @@
-wasm-pack build --target web sbv2_wasm
+#!/bin/sh
-wasm-opt -O3 -o ./sbv2_wasm/pkg/sbv2_wasm_bg.wasm ./sbv2_wasm/pkg/sbv2_wasm_bg.wasm
+wasm-pack build --target web ./crates/sbv2_wasm --release
-wasm-strip ./sbv2_wasm/pkg/sbv2_wasm_bg.wasm
+wasm-opt -O3 -o ./crates/sbv2_wasm/pkg/sbv2_wasm_bg.wasm ./crates/sbv2_wasm/pkg/sbv2_wasm_bg.wasm
-mkdir -p ./sbv2_wasm/dist
+wasm-strip ./crates/sbv2_wasm/pkg/sbv2_wasm_bg.wasm
-cp ./sbv2_wasm/sbv2_wasm/pkg/sbv2_wasm_bg.wasm ./sbv2_wasm/dist/sbv2_wasm_bg.wasm
+mkdir -p ./crates/sbv2_wasm/dist
+cp ./crates/sbv2_wasm/pkg/sbv2_wasm_bg.wasm ./crates/sbv2_wasm/dist/sbv2_wasm_bg.wasm
+cd ./crates/sbv2_wasm
+pnpm build

crates/sbv2_wasm/example.js

@@ -3,9 +3,12 @@ import fs from "node:fs/promises";
 
 ModelHolder.globalInit(await fs.readFile("./dist/sbv2_wasm_bg.wasm"));
 const holder = await ModelHolder.create(
-	(await fs.readFile("../models/tokenizer.json")).toString("utf-8"),
+	(await fs.readFile("../../models/tokenizer.json")).toString("utf-8"),
-	await fs.readFile("../models/deberta.onnx"),
+	await fs.readFile("../../models/deberta.onnx"),
+);
+await holder.load(
+	"tsukuyomi",
+	await fs.readFile("../../models/tsukuyomi.sbv2"),
 );
-await holder.load("tsukuyomi", await fs.readFile("../models/iroha2.sbv2"));
 await fs.writeFile("out.wav", await holder.synthesize("tsukuyomi", "おはよう"));
 holder.unload("tsukuyomi");

crates/sbv2_wasm/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "sbv2",
-	"version": "0.1.1",
+	"version": "0.2.0-alpha6",
 	"description": "Style Bert VITS2 wasm",
 	"main": "dist/index.js",
 	"types": "dist/index.d.ts",
@@ -21,9 +21,5 @@
 	"dependencies": {
 		"onnxruntime-web": "^1.20.1"
 	},
-	"files": [
+	"files": ["dist/*", "package.json", "README.md", "pkg/*.ts", "pkg/*.js"]
-		"dist/*",
-		"package.json",
-		"README.md"
-	]
 }

crates/sbv2_wasm/src-js/index.ts

@@ -74,6 +74,8 @@ export class ModelHolder {
 							style_vec: e,
 							sdp_ratio: new Tensor("float32", [f]),
 							length_scale: new Tensor("float32", [g]),
+							noise_scale: new Tensor("float32", [0.677]),
+							noise_scale_w: new Tensor("float32", [0.8]),
 						})
 					).output;
 					return [new Uint32Array(res.dims), await res.getData(true)];