mirror of
https://github.com/geekwenjie/SmartJavaAI.git
synced 2026-09-14 05:38:50 +00:00
nllp机器翻译支持202种语言互相翻译
(cherry picked from commit cb0cfff2863cf1c29132de76716c676be24bcad2)
This commit is contained in:
@@ -0,0 +1,35 @@
|
||||
package cn.smartjavaai.common.pool;
|
||||
|
||||
import ai.djl.inference.Predictor;
|
||||
import ai.djl.repository.zoo.ZooModel;
|
||||
import org.apache.commons.pool2.BasePooledObjectFactory;
|
||||
import org.apache.commons.pool2.PooledObject;
|
||||
import org.apache.commons.pool2.impl.DefaultPooledObject;
|
||||
|
||||
/**
|
||||
* ZooModel 工厂类
|
||||
* @author lwx
|
||||
* @date 2025/6/06
|
||||
*/
|
||||
public class ZooModelFactory<I, O> extends BasePooledObjectFactory<ZooModel<I, O>> {
|
||||
private final ZooModel<I, O> model;
|
||||
|
||||
public ZooModelFactory(ZooModel<I, O> model) {
|
||||
this.model = model;
|
||||
}
|
||||
|
||||
@Override
|
||||
public ZooModel<I, O> create() {
|
||||
return model;
|
||||
}
|
||||
|
||||
@Override
|
||||
public PooledObject<ZooModel<I, O>> wrap(ZooModel<I, O> predictor) {
|
||||
return new DefaultPooledObject<>(predictor);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void destroyObject(PooledObject<ZooModel<I, O>> p) {
|
||||
p.getObject().close();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user