arthur/xiao
1
0
Fork 0
mirror of https://github.com/arthur-pbty/xiao.git synced 2026-06-03 23:26:21 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
Dragon Fire
2024-05-09 14:58:17 -04:00
parent f102b72b24
commit cd4dc65420
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
structures/Tensorflow.js
+1 -1
View File
@@ -83,7 +83,7 @@ module.exports = class Tensorflow {
const stylizedImage = (await this.transformerModel.predict([loadedImage, stylePrediction])).squeeze();
loadedImage.dispose();
stylePrediction.dispose();
const encodedImg = stylizedImage.mul(255).toInt();
const encodedImg = stylizedImage.mul(255).clipByValue(0, 255).toInt();
stylizedImage.dispose();
const buffer = await tf.node.encodeJpeg(encodedImg);
encodedImg.dispose();