fix: Allow marian configs to deserialize from json. (#2556)

huggingface · Oct 13, 2024 · 41ade77 · 41ade77
1 parent 6eab6b5
commit 41ade77
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/candle-transformers/src/models/marian.rs b/candle-transformers/src/models/marian.rs
@@ -1,8 +1,9 @@
 use super::with_tracing::{linear, Embedding, Linear};
 use candle::{Result, Tensor};
 use candle_nn::{layer_norm, LayerNorm, VarBuilder};
+use serde::Deserialize;
 
-#[derive(Debug, Clone)]
+#[derive(Debug, Clone, Deserialize)]
 pub struct Config {
  pub vocab_size: usize,
  pub decoder_vocab_size: Option<usize>,