diff --git a/wasm/inference-models/.cargo/config.toml b/wasm/.cargo/config.toml
similarity index 78%
rename from wasm/inference-models/.cargo/config.toml
rename to wasm/.cargo/config.toml
index 82e6a7f..3820420 100644
--- a/wasm/inference-models/.cargo/config.toml
+++ b/wasm/.cargo/config.toml
@@ -3,5 +3,6 @@ rustflags = [
     "-Ctarget-feature=+simd128,+bulk-memory,+reference-types",
     '--cfg', 'getrandom_backend="wasm_js"',
     "--cfg=has_std",
-    "-Zthreads=6"
+    "-Zlocation-detail=none",
+    "-Zthreads=6",
 ]
diff --git a/wasm/Cargo.toml b/wasm/Cargo.toml
index eef176b..8a12cf2 100644
--- a/wasm/Cargo.toml
+++ b/wasm/Cargo.toml
@@ -1,13 +1,16 @@
 [workspace]
 resolver = "3"
-rust-version = "1.94"
-edition = "2024"
 members = ["inference-models", "model-ai-text"]
 
+[workspace.package]
+version = "0.0.1-rc.1"
+rust-version = "1.96"
+edition = "2024"
+
 [workspace.dependencies]
 ## Workspace models
 model-ai-text.path = "./model-ai-text"
-model-ai-text.version = "^0.1"
+model-ai-text.version = "=0.0.1-rc.1"
 ## Wasm
 wasm-bindgen.version = "=0.2.114"
 wasm-bindgen-futures.version = "^0.4"
@@ -32,9 +35,11 @@ tokenizers.features = ["unstable_wasm"]
 burn.git = "https://github.com/tracel-ai/burn.git"
 burn.rev = "3a0e604cb14e445cd5a3686fbe06d0e11c76f05a"
 burn.default-features = false
+burn.features = ["std"]
 burn-store.git = "https://github.com/tracel-ai/burn.git"
 burn-store.rev = "3a0e604cb14e445cd5a3686fbe06d0e11c76f05a"
 burn-store.default-features = false
+burn-store.features = ["std"]
 burn-onnx.git = "https://github.com/tracel-ai/burn-onnx.git"
 burn-onnx.rev = "17a0864e02c19669b143e4227038363bb5194589"
 burn-onnx.default-features = false
diff --git a/wasm/inference-models/Cargo.toml b/wasm/inference-models/Cargo.toml
index 7a56efd..6d1b016 100644
--- a/wasm/inference-models/Cargo.toml
+++ b/wasm/inference-models/Cargo.toml
@@ -1,7 +1,8 @@
 [package]
 name = "inference-models"
-version = "0.1.0"
-edition = "2024"
+version.workspace = true
+edition.workspace = true
+rust-version.workspace = true
 
 [lib]
 crate-type = ["cdylib", "rlib"]
@@ -21,7 +22,7 @@ console_error_panic_hook.workspace = true
 tracing.workspace = true
 ## Inf
 burn.workspace = true
-burn.features = ["ndarray", "webgpu"]
+burn.features = ["ndarray", "wgpu", "webgpu"]
 burn-store.workspace = true
 burn-store.features = ["burnpack"]
 tokenizers.workspace = true
diff --git a/wasm/model-ai-text/Cargo.toml b/wasm/model-ai-text/Cargo.toml
index 3811c53..eb227f0 100644
--- a/wasm/model-ai-text/Cargo.toml
+++ b/wasm/model-ai-text/Cargo.toml
@@ -1,7 +1,8 @@
 [package]
 name = "model-ai-text"
-version = "0.1.0"
-edition = "2024"
+version.workspace = true
+edition.workspace = true
+rust-version.workspace = true
 
 [dependencies]
 ## Logging