huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 03:28:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Mikarific
2024-10-13 15:05:50 -06:00
committed by GitHub
parent 6eab6b57f5
commit 41ade774e8
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
candle-transformers/src/models/marian.rs
View File

@ -1,8 +1,9 @@
use super::with_tracing::{linear, Embedding, Linear}; use super::with_tracing::{linear, Embedding, Linear};
use candle::{Result, Tensor}; use candle::{Result, Tensor};
use candle_nn::{layer_norm, LayerNorm, VarBuilder}; use candle_nn::{layer_norm, LayerNorm, VarBuilder};
use serde::Deserialize;
#[derive(Debug, Clone)] #[derive(Debug, Clone, Deserialize)]
pub struct Config { pub struct Config {
pub vocab_size: usize, pub vocab_size: usize,
pub decoder_vocab_size: Option<usize>, pub decoder_vocab_size: Option<usize>,