Improve log
This commit is contained in:
parent
1214b4ddc3
commit
7dccdecff3
|
@ -64,7 +64,10 @@ const logInfo = (...params: any[]) => {
|
||||||
};
|
};
|
||||||
|
|
||||||
const logDebug = (param: () => any) => {
|
const logDebug = (param: () => any) => {
|
||||||
if (isDev) console.log(`[debug] ${util.inspect(param())}`);
|
if (isDev) {
|
||||||
|
const p = param();
|
||||||
|
console.log(`[debug] ${typeof p == "string" ? p : util.inspect(p)}`);
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|
|
@ -172,10 +172,9 @@ const clipImageEmbedding_ = async (jpegFilePath: string) => {
|
||||||
};
|
};
|
||||||
const t2 = Date.now();
|
const t2 = Date.now();
|
||||||
const results = await imageSession.run(feeds);
|
const results = await imageSession.run(feeds);
|
||||||
log.info(
|
log.debug(
|
||||||
`onnx image embedding time: ${Date.now() - t1} ms (prep:${
|
() =>
|
||||||
t2 - t1
|
`CLIP image embedding took ${Date.now() - t1} ms (prep: ${t2 - t1} ms, inference: ${Date.now() - t2} ms)`,
|
||||||
} ms, extraction: ${Date.now() - t2} ms)`,
|
|
||||||
);
|
);
|
||||||
const imageEmbedding = results["output"].data; // Float32Array
|
const imageEmbedding = results["output"].data; // Float32Array
|
||||||
return normalizeEmbedding(imageEmbedding);
|
return normalizeEmbedding(imageEmbedding);
|
||||||
|
@ -280,10 +279,9 @@ export const clipTextEmbedding = async (text: string) => {
|
||||||
};
|
};
|
||||||
const t2 = Date.now();
|
const t2 = Date.now();
|
||||||
const results = await imageSession.run(feeds);
|
const results = await imageSession.run(feeds);
|
||||||
log.info(
|
log.debug(
|
||||||
`onnx text embedding time: ${Date.now() - t1} ms (prep:${
|
() =>
|
||||||
t2 - t1
|
`CLIP text embedding took ${Date.now() - t1} ms (prep: ${t2 - t1} ms, inference: ${Date.now() - t2} ms)`,
|
||||||
} ms, extraction: ${Date.now() - t2} ms)`,
|
|
||||||
);
|
);
|
||||||
const textEmbedding = results["output"].data;
|
const textEmbedding = results["output"].data;
|
||||||
return normalizeEmbedding(textEmbedding);
|
return normalizeEmbedding(textEmbedding);
|
||||||
|
|
Loading…
Reference in a new issue