ai.djl.repository.zoo

Class DefaultModelZoo

java.lang.Object

ai.djl.repository.zoo.ModelZoo

ai.djl.repository.zoo.DefaultModelZoo

public class DefaultModelZoo
extends 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() Constructs a new LocalModelZoo instance.
DefaultModelZoo(java.lang.String locations) Constructs a new 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 ai.djl.repository.zoo.ModelZoo

getModelLoader, getModelZoo, hasModelZoo, listModels, listModels, listModelZoo, loadModel, registerModelZoo

Methods inherited from class java.lang.Object

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

Field Detail

GROUP_ID

public static final java.lang.String GROUP_ID

See Also:

Constant Field Values

Constructor Detail

DefaultModelZoo

public DefaultModelZoo()

Constructs a new LocalModelZoo instance.

DefaultModelZoo

public DefaultModelZoo(java.lang.String locations)

Constructs a new 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 class 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 class 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 class ModelZoo

Returns:

all supported engine names