Class DjlAutoConfiguration

java.lang.Object
ai.djl.spring.configuration.DjlAutoConfiguration

@Configuration @ConditionalOnMissingBean(ai.djl.repository.zoo.ZooModel.class) @EnableConfigurationProperties(DjlConfigurationProperties.class) public class DjlAutoConfiguration extends Object
  • Constructor Details

    • DjlAutoConfiguration

      public DjlAutoConfiguration()
  • Method Details

    • model

      @Bean public ai.djl.repository.zoo.ZooModel<?,?> model() throws ai.djl.MalformedModelException, ai.djl.repository.zoo.ModelNotFoundException, IOException
      Throws:
      ai.djl.MalformedModelException
      ai.djl.repository.zoo.ModelNotFoundException
      IOException
    • predictorProvider

      @Bean public Supplier<ai.djl.inference.Predictor<?,?>> predictorProvider(ai.djl.repository.zoo.ZooModel<?,?> model)
      Expected to be used with try-with-resources. The provided predictor is AutoCloseable.
      Parameters:
      model - injected configured model
      Returns:
      provider of the predictor object