Package 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	Description
static final String	GROUP_ID

Constructor Summary

Constructors

Constructor	Description
DefaultModelZoo()	Constructs a new LocalModelZoo instance.
DefaultModelZoo(String locations)	Constructs a new LocalModelZoo instance from the given search locations.

Method Summary

Modifier and Type	Method	Description
String	getGroupId()	Returns the global unique identifier of the ModelZoo.
Set<String>	getSupportedEngines()	Returns all supported engine names.

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

addModel, addModel, getModelLoader, getModelLoaders, getModelZoo, hasModelZoo, listModels, listModels, listModelZoo, loadModel, registerModelZoo, setModelZooResolver

Methods inherited from class java.lang.Object

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

Field Details

GROUP_ID

public static final String GROUP_ID

See Also:

• Constant Field Values

Constructor Details

DefaultModelZoo

public DefaultModelZoo()

Constructs a new LocalModelZoo instance.

DefaultModelZoo

public DefaultModelZoo(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 Details

getGroupId

public 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 Set<String> getSupportedEngines()

Returns all supported engine names.

Specified by:

getSupportedEngines in class ModelZoo

Returns:

all supported engine names