diff --git a/project/Cargo.toml b/project/Cargo.toml
index ddcc56e..6c19785 100644
--- a/project/Cargo.toml
+++ b/project/Cargo.toml
@@ -4,4 +4,6 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
-burn = { version = "0.16.0", features = ["webgpu", "train", "vision"] }
+burn = { version = "0.16.0", features = ["wgpu", "train", "vision"] }
+log = { version = "0.4.26" }
+serde = { version = "1.0.219", features = ["std", "derive"] }