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

This commit is contained in:
Mikarific
2024-10-13 15:05:50 -06:00
committed by GitHub
parent 6eab6b57f5
commit 41ade774e8

View File

@ -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>,