ai.djl.repository.zoo

Class DefaultModelZoo

java.lang.Object

ai.djl.repository.zoo.DefaultModelZoo

All Implemented Interfaces:

ModelZoo

public class DefaultModelZoo
extends java.lang.Object
implements ModelZoo

A ModelZoo that contains models in specified locations.

Field Summary

Fields

Modifier and Type	Field and Description
static java.lang.String	GROUP_ID

Constructor Summary

Constructors

Constructor and Description
DefaultModelZoo(java.lang.String locations) Creates the LocalModelZoo instance from the given search locations.

Method Summary

Modifier and Type	Method and Description
java.lang.String	getGroupId() Returns the global unique identifier of the ModelZoo.
java.util.List<ModelLoader>	getModelLoaders() Lists the available model families in the ModelZoo.
java.util.Set<java.lang.String>	getSupportedEngines() Returns all supported engine names.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface ai.djl.repository.zoo.ModelZoo

getModelLoader, listModels, listModels, loadModel

Field Detail

GROUP_ID

public static final java.lang.String GROUP_ID

See Also:

Constant Field Values

Constructor Detail

DefaultModelZoo

public DefaultModelZoo(java.lang.String locations)

Creates the LocalModelZoo instance from the given search locations.

Parameters:

locations - a comma separated urls where the models to be loaded from

Method Detail

getModelLoaders

public java.util.List<ModelLoader> getModelLoaders()

Lists the available model families in the ModelZoo.

Specified by:

getModelLoaders in interface ModelZoo

Returns:

the list of all available model families

getGroupId

public java.lang.String getGroupId()

Returns the global unique identifier of the ModelZoo.

We recommend to use reverse DNS name as your model zoo group ID to make sure it's not conflict with other ModelZoos.

Specified by:

getGroupId in interface ModelZoo

Returns:

the global unique identifier of the ModelZoo

getSupportedEngines

public java.util.Set<java.lang.String> getSupportedEngines()

Returns all supported engine names.

Specified by:

getSupportedEngines in interface ModelZoo

Returns:

all supported engine names