ai.djl.repository.zoo

Class LocalModelZoo

java.lang.Object

ai.djl.repository.zoo.LocalModelZoo

All Implemented Interfaces:

ModelZoo

public class LocalModelZoo
extends java.lang.Object
implements ModelZoo

A ModelZoo that contains models in local directory.

Field Summary

Fields

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

Constructor Summary

Constructors

Constructor and Description
LocalModelZoo(java.nio.file.Path folder) Creates the LocalModelZoo instance from the given directory.

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, loadModel

Field Detail

GROUP_ID

public static final java.lang.String GROUP_ID

See Also:

Constant Field Values

Constructor Detail

LocalModelZoo

public LocalModelZoo(java.nio.file.Path folder)

Creates the LocalModelZoo instance from the given directory.

Parameters:

folder - the directory to load models 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