arthur/xiao
1
0
Fork 0
mirror of https://github.com/arthur-pbty/xiao.git synced 2026-06-18 13:56:25 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
Dragon Fire
2024-05-08 18:22:33 -04:00
parent 8a8c795e5f
commit f40ded5a85
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
@@ -80,7 +80,7 @@ module.exports = class Tensorflow {
styleTensor.dispose(); styleTensor.dispose();
const stylePrediction = await this.styleModel.predict(loadedStyle); const stylePrediction = await this.styleModel.predict(loadedStyle);
loadedStyle.dispose(); loadedStyle.dispose();
const stylizedImage = await this.transformerModel.predict([loadedImage, stylePrediction.squeeze()]); const stylizedImage = await this.transformerModel.predict([loadedImage, stylePrediction]);
loadedImage.dispose(); loadedImage.dispose();
stylePrediction.dispose(); stylePrediction.dispose();
const buffer = await tf.node.encodePng(stylizedImage.squeeze()); const buffer = await tf.node.encodePng(stylizedImage.squeeze());