File tree Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Expand file tree Collapse file tree 1 file changed +3
-4
lines changed Original file line number Diff line number Diff line change @@ -59,9 +59,8 @@ enum Config {
5959 NomicBert ( NomicConfig ) ,
6060 #[ allow( dead_code) ]
6161 Mistral ( MistralConfig ) ,
62+ #[ serde( alias = "new" ) ]
6263 Gte ( GTEConfig ) ,
63- #[ serde( rename = "new" ) ]
64- GteAlibaba ( GTEConfig ) ,
6564 #[ allow( dead_code) ]
6665 Qwen2 ( Qwen2Config ) ,
6766 #[ serde( rename = "mpnet" ) ]
@@ -224,7 +223,7 @@ impl CandleBackend {
224223 "Mistral is only supported on Cuda devices in fp16 with flash attention enabled"
225224 . to_string ( ) ,
226225 ) ) ,
227- ( Config :: Gte ( config) | Config :: GteAlibaba ( config ) , Device :: Cpu | Device :: Metal ( _) ) => {
226+ ( Config :: Gte ( config) , Device :: Cpu | Device :: Metal ( _) ) => {
228227 tracing:: info!( "Starting GTE model on {:?}" , device) ;
229228 Ok ( Box :: new ( GTEModel :: load ( vb, & config, model_type) . s ( ) ?) )
230229 }
@@ -355,7 +354,7 @@ impl CandleBackend {
355354 ) )
356355 }
357356 #[ cfg( feature = "cuda" ) ]
358- ( Config :: Gte ( config) | Config :: GteAlibaba ( config ) , Device :: Cuda ( _) ) => {
357+ ( Config :: Gte ( config) , Device :: Cuda ( _) ) => {
359358 if dtype != DType :: F16
360359 || !cfg ! ( any( feature = "flash-attn" , feature = "flash-attn-v1" ) )
361360 {
You can’t perform that action at this time.
0 commit comments