yizhi-js-lib/ai-box
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

修复模型下载问题

Browse Source
This commit is contained in:
Yizhi
2025-03-10 15:34:24 +08:00
parent 2dfe063049
commit 92e46c2c33
1 changed files with 29 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
src/deploy/common/model.ts
View File

@ -68,7 +68,7 @@ export abstract class Model {
} }
//不存在则下载 //不存在则下载
let cache = cacheJsonData.find(c => c.url === url); let cache = cacheJsonData.find(c => c.url === url);
if (!cache) { if (!cache || !fs.existsSync(path.join(cacheDir, cache.filename))) {
let saveType = option?.saveType ?? null; let saveType = option?.saveType ?? null;
const saveTypeDict: Record<string, ModelType> = { const saveTypeDict: Record<string, ModelType> = {
".onnx": "onnx", ".onnx": "onnx",
@ -83,33 +83,40 @@ export abstract class Model {
return res.blob(); return res.blob();
}).then(blob => blob.stream()).then(async stream => { }).then(blob => blob.stream()).then(async stream => {
const cacheFilename = path.join(cacheDir, Date.now().toString()); const cacheFilename = path.join(cacheDir, Date.now().toString());
let fsStream!: ReturnType<typeof fs.createWriteStream>; const hash = await new Promise<string>((resolve, reject) => {
let hashStream!: ReturnType<typeof crypto.createHash>; let fsStream!: ReturnType<typeof fs.createWriteStream>;
let hash!: string; let hashStream!: ReturnType<typeof crypto.createHash>;
await stream.pipeTo(new WritableStream({ stream.pipeTo(new WritableStream({
start(controller) { start(controller) {
fsStream = fs.createWriteStream(cacheFilename); fsStream = fs.createWriteStream(cacheFilename);
hashStream = crypto.createHash("md5"); hashStream = crypto.createHash("md5");
}, },
async write(chunk, controller) { async write(chunk, controller) {
await new Promise<void>((resolve, reject) => fsStream.write(chunk, err => err ? reject(err) : resolve())); await new Promise<void>((resolve, reject) => fsStream.write(chunk, err => err ? reject(err) : resolve()));
await new Promise<void>((resolve, reject) => hashStream.write(chunk, err => err ? reject(err) : resolve())); await new Promise<void>((resolve, reject) => hashStream.write(chunk, err => err ? reject(err) : resolve()));
}, },
close() { close() {
fsStream.end(); fsStream.end();
hashStream.end(); hashStream.end();
hash = hashStream.digest("hex") resolve(hashStream.digest("hex"));
}, },
abort() { } abort() { }
})); })).catch(reject);
})
return { filename: cacheFilename, hash }; return { filename: cacheFilename, hash };
}); });
//重命名 //重命名
const filename = `${res.hash}.${saveType}`; const filename = `${res.hash}.${saveType}`;
fs.promises.rename(res.filename, path.join(cacheDir, filename)); fs.promises.rename(res.filename, path.join(cacheDir, filename));
//保存缓存 //保存缓存
cache = { url, filename }; if (!cache) {
cacheJsonData.push(cache); cache = { url, filename };
cacheJsonData.push(cache);
}
else {
cache.filename = filename;
cache.url = url;
}
fs.promises.writeFile(cacheJsonFile, JSON.stringify(cacheJsonData, null, 4)); fs.promises.writeFile(cacheJsonFile, JSON.stringify(cacheJsonData, null, 4));
} }
//返回模型数据 //返回模型数据