@@ -338,10 +338,17 @@ async fn init_backend(
338
338
{
339
339
if let Some ( api_repo) = api_repo. as_ref ( ) {
340
340
let start = std:: time:: Instant :: now ( ) ;
341
- download_onnx ( api_repo)
341
+ let model_files = download_onnx ( api_repo)
342
342
. await
343
- . map_err ( |err| BackendError :: WeightsNotFound ( err. to_string ( ) ) ) ;
344
- tracing:: info!( "Model ONNX weights downloaded in {:?}" , start. elapsed( ) ) ;
343
+ . map_err ( |err| BackendError :: WeightsNotFound ( err. to_string ( ) ) ) ?;
344
+ match model_files. is_empty ( ) {
345
+ true => {
346
+ tracing:: error!( "Model ONNX files not found in the repository" )
347
+ }
348
+ false => {
349
+ tracing:: info!( "Model ONNX weights downloaded in {:?}" , start. elapsed( ) )
350
+ }
351
+ }
345
352
}
346
353
347
354
let backend = OrtBackend :: new ( & model_path, dtype. to_string ( ) , model_type. clone ( ) ) ;
@@ -530,8 +537,11 @@ async fn download_onnx(api: &ApiRepo) -> Result<Vec<PathBuf>, ApiError> {
530
537
Err ( err) => {
531
538
tracing:: warn!( "Could not download `model.onnx`: {err}" ) ;
532
539
tracing:: info!( "Downloading `onnx/model.onnx`" ) ;
533
- let p = api. get ( "onnx/model.onnx" ) . await ?;
534
- model_files. push ( p. parent ( ) . unwrap ( ) . to_path_buf ( ) )
540
+
541
+ match api. get ( "onnx/model.onnx" ) . await {
542
+ Ok ( p) => model_files. push ( p. parent ( ) . unwrap ( ) . to_path_buf ( ) ) ,
543
+ Err ( err) => tracing:: warn!( "Could not download `onnx/model.onnx`: {err}" ) ,
544
+ } ;
535
545
}
536
546
} ;
537
547
0 commit comments