From 2feb0b054f96e4c4c87f01b243e749896c94f8c7 Mon Sep 17 00:00:00 2001 From: Laurent Mazare Date: Wed, 8 Nov 2023 08:23:53 +0100 Subject: [PATCH] Add the mel filters for 128 bins. (#1295) --- candle-examples/examples/whisper/main.rs | 6 +++++- .../examples/whisper/melfilters128.bytes | Bin 0 -> 102912 bytes 2 files changed, 5 insertions(+), 1 deletion(-) create mode 100644 candle-examples/examples/whisper/melfilters128.bytes diff --git a/candle-examples/examples/whisper/main.rs b/candle-examples/examples/whisper/main.rs index f2560815..d2caebcd 100644 --- a/candle-examples/examples/whisper/main.rs +++ b/candle-examples/examples/whisper/main.rs @@ -525,7 +525,11 @@ fn main() -> Result<()> { let config: Config = serde_json::from_str(&std::fs::read_to_string(config_filename)?)?; let tokenizer = Tokenizer::from_file(tokenizer_filename).map_err(E::msg)?; - let mel_bytes = include_bytes!("melfilters.bytes"); + let mel_bytes = match config.num_mel_bins { + 80 => include_bytes!("melfilters.bytes").as_slice(), + 128 => include_bytes!("melfilters128.bytes").as_slice(), + nmel => anyhow::bail!("unexpected num_mel_bins {nmel}"), + }; let mut mel_filters = vec![0f32; mel_bytes.len() / 4]; ::read_f32_into(mel_bytes, &mut mel_filters); diff --git a/candle-examples/examples/whisper/melfilters128.bytes b/candle-examples/examples/whisper/melfilters128.bytes new file mode 100644 index 0000000000000000000000000000000000000000..f287c5b1dda3e2cfb3952d2ee1d6d65fa6b9c8f6 GIT binary patch literal 102912 zcmeI*YgCTu8VB&#OlfN(N^Mb=tPmrGRPXb@S#%&Gr!j~a<9wV6Lo;M3MxwMS8>uGe zLq~@uhZc<*G=`)YgBpjhhiS7$jfOE>dp_#DpT6IJUs%ue_;CH!|6135=T!!SVWr=x zA;Q346rg}ffwH<6oDq{cEKuNY0l!nloD+tD0uq5ENt-z%5+9Z*P%qG{d7HkJpa2C- z3k1~Wp?~Bpn0|v53Q*ukk}Y=$V?Y5Nft(dPcsJ28Vp(kgzZ(Ud5w#=0?i(R6yV}4r zS0kMGin;;;wLan(_c%>m0d`ISeSzgCx^b82Gw}_@1^P!$&WN0d zRSFmeq+aEmFb5Qn2pr9e;fzRpSfW6^K+XzB-rpGo3P=Qk<|M+&!oYnZF=B}Vp53gu zOBe$R=m=y_*~49;W5lxB0&hM)&lyoW0_?sK0{@w8&NEjdocM~m0)suey!k9w3;tg$ zbtAye8zJy2urqf_Bb@k(+5#eQ0q-ViM}Xby2wdy^kTarV#4-ime2)mu31dJ3i9l}o zD$a<+hb0Qs3wS*3!23I+KmmzB+Nr5%_d_f06NwQ^6gZk^$z8%2P(Vi@%RiI5M8}9_ zwFL&*W^+c=jsUxFgutvT&3NW&gcDy;S0HGPov0WzhM(n9Hv;Uu5dsm{ow-XI;lx+e z7I3my$-9Z#5n%T^0zZzZ;Ed=Pu}pz})2lcqi~$8C0t>3bI3p4tmMBm!kUgan@9&HP z1tbE)%pLK&YZvYli4jW_FkjY$yM!^IfR4cT_7}KIbc|S5TVQCHWdh@V;?KoWI|l4t zN5C`8o@Xu{BbF(UzH>LvT#Nw)Bm)0E93#Ha`tdW55+jx<5FXu>XD-Ho0y+ZiGtThL zrDMdh+5*|PJjJw=?wk{~W5Dip1ajM*;Ed=Pu}py@US_5nRCJ@P(UJ(oYr0I9(a~>A~9l#0#zAP;8dBxIbjqiAQ3o{n1%5jQ#mIR zBbF!-TvP_fNn1E4i~?}6w(2_p2-Y2HyJPApL%Jt;&S3U}q%i&3C}M4;p& zD{-w)n0OT$&9j%pi6ylK-uW~fbCzxpt7f0)*-PyxuzMW=%Y*$z`?+2`d+9i_thPYz zAEj`~jS>ZqVz^V(jsm-v2owY)YwMeg6q~!W=T4D0u|xrjy${jr!eS8;ypB7CaiD++ zfoZYdY10e*#qtnm?i3S7d`y9Qfnhc0urqj>7@eQXox(^^z=XiD)&)j~t{p|%yP7DO z-;}dr!ikS5V08Wx58^%$iLPcme=!aekO*uwyN!$1Q$@&!Iov4{CzjL}2#)vF)()_J z^RrPj;n0WQ&tB~`uzQ(6h?x=M&mLlPTw7u3^pKzZl3}r?v_S05jWBNa6*m&z5l%zg zI4er0fz8VVlB|xx=}EAto_UXV6&Vz3$_tFxRg7LwMvL^&OPm$u(?9@I0+G$ndJncP zM`F!jQIb)}Suq90=gJG1&)Q=w-BpIWfdL}azKFA;d?E;7O5l!Lvhmw(#kd_DAoeY| zg#E8t@osMli_eu7xbYI+zOC}$ch_5_?5TtZ%Hgaiod`BB6IgDWf`ItW!t2cMIG*po zbC?W^HKhfHxGqCVPaE;i5It;GQG7jh3x<2g)Mj4!Av z5L@SpB8v(<@7h+BTW6xVdn$K}s$pQ?Qh}{m0h-;pBsk8q7GEEFjBAU=a<@puSX5cS zzQ7%oZ+=ew$T4=J&FO4-1>fXuQ8^Io-K4;Q#r9f6={oeDYY=|(Y7x*c5*y6!7)|~K z7AT;wz`#9qT2^Tmnx?cANA~@IPWSurET(WC*t%RG^|$@cl-fvy4=J5n`?1(@L|r2qIqE3a)B=fp49$Py%u4X zcQB#Z&j`0#hv0taIWKZHRuvVXj@#vNJob#e+9@w^AAauq$ z?OgN>c(@m0q4#6-)OMiq^S1o@n4FDOMFm!DcGgOgeDTrjOc-2V;95mG#`X4t&F)^@ zF^Xn_ZJQSOD{6#~hnGEE6BFQLU4fUg3eauRBAh5Y$9XZ$#)`rM(_R*8(O)e<=Ae9B z8Fm9po+rTD%NdrAh5Ss6!nt7U4GXM%?X1N=b;IfGZSa|W3%O4YWBQh}@hWQCuL%E=GH{$_FX065*eB84i~F@qXxJ3>|u#J4W$r zu>FPwhTlJ_W!D8^hanjQnwG+H)D9fLFof?t%b97Ij87F7sD4@Fy*{Q~+mtdIPmzKt zb1qo_Py3BQ^8%+Fdd&NED-qN7R^1~0h-$^bR3%mU;mv5dax9>{_i8tFJE*1b0hCH u3g?5Zzb)|1j5_0>->+(6#h&Q5?Ms+@?tx|VP1tBV;mz+h^?&~VuK6Ejlbd+} literal 0 HcmV?d00001