Merge pull request #28 from tracel-ai/fix/bert-0.13.0

Remove explicit fusion backend in Bert example
tracel-ai · Apr 29, 2024 · a00bee0 · a00bee0
2 parents 29fadae + de3a736
commit a00bee0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/bert-burn/Cargo.toml b/bert-burn/Cargo.toml
@@ -2,7 +2,7 @@
 authors = ["Aasheesh Singh [email protected]"]
 license = "MIT OR Apache-2.0"
 name = "bert-burn"
-version = "0.1.0"
+version = "0.2.0"
 edition = "2021"
 
 [features]

diff --git a/bert-burn/examples/infer-embedding.rs b/bert-burn/examples/infer-embedding.rs
@@ -133,10 +133,9 @@ mod tch_cpu {
 mod wgpu {
     use crate::{launch, ElemType};
     use burn::backend::wgpu::{AutoGraphicsApi, Wgpu, WgpuDevice};
-    use burn::backend::Fusion;
 
     pub fn run() {
-        launch::<Fusion<Wgpu<AutoGraphicsApi, ElemType, i32>>>(WgpuDevice::default());
+        launch::<Wgpu<AutoGraphicsApi, ElemType, i32>>(WgpuDevice::default());
     }
 }