From 22ee2a549780c4129697995c18bb2094e3105b60 Mon Sep 17 00:00:00 2001 From: Dragon Fire Date: Mon, 13 May 2024 23:16:15 -0400 Subject: [PATCH] Move Tensorflow models into assets --- .../tensorflow}/style_js/group1-shard1of9 | Bin .../tensorflow}/style_js/group1-shard2of9 | Bin .../tensorflow}/style_js/group1-shard3of9 | Bin .../tensorflow}/style_js/group1-shard4of9 | Bin .../tensorflow}/style_js/group1-shard5of9 | Bin .../tensorflow}/style_js/group1-shard6of9 | Bin .../tensorflow}/style_js/group1-shard7of9 | Bin .../tensorflow}/style_js/group1-shard8of9 | Bin .../tensorflow}/style_js/group1-shard9of9 | Bin .../tensorflow}/style_js/model.json | 0 .../tensorflow}/transformer_js/group1-shard1of2 | Bin .../tensorflow}/transformer_js/group1-shard2of2 | Bin .../tensorflow}/transformer_js/model.json | 0 structures/Tensorflow.js | 4 ++-- 14 files changed, 2 insertions(+), 2 deletions(-) rename {tf_models => assets/tensorflow}/style_js/group1-shard1of9 (100%) rename {tf_models => assets/tensorflow}/style_js/group1-shard2of9 (100%) rename {tf_models => assets/tensorflow}/style_js/group1-shard3of9 (100%) rename {tf_models => assets/tensorflow}/style_js/group1-shard4of9 (100%) rename {tf_models => assets/tensorflow}/style_js/group1-shard5of9 (100%) rename {tf_models => assets/tensorflow}/style_js/group1-shard6of9 (100%) rename {tf_models => assets/tensorflow}/style_js/group1-shard7of9 (100%) rename {tf_models => assets/tensorflow}/style_js/group1-shard8of9 (100%) rename {tf_models => assets/tensorflow}/style_js/group1-shard9of9 (100%) rename {tf_models => assets/tensorflow}/style_js/model.json (100%) rename {tf_models => assets/tensorflow}/transformer_js/group1-shard1of2 (100%) rename {tf_models => assets/tensorflow}/transformer_js/group1-shard2of2 (100%) rename {tf_models => assets/tensorflow}/transformer_js/model.json (100%) diff --git a/tf_models/style_js/group1-shard1of9 b/assets/tensorflow/style_js/group1-shard1of9 similarity index 100% rename from tf_models/style_js/group1-shard1of9 rename to assets/tensorflow/style_js/group1-shard1of9 diff --git a/tf_models/style_js/group1-shard2of9 b/assets/tensorflow/style_js/group1-shard2of9 similarity index 100% rename from tf_models/style_js/group1-shard2of9 rename to assets/tensorflow/style_js/group1-shard2of9 diff --git a/tf_models/style_js/group1-shard3of9 b/assets/tensorflow/style_js/group1-shard3of9 similarity index 100% rename from tf_models/style_js/group1-shard3of9 rename to assets/tensorflow/style_js/group1-shard3of9 diff --git a/tf_models/style_js/group1-shard4of9 b/assets/tensorflow/style_js/group1-shard4of9 similarity index 100% rename from tf_models/style_js/group1-shard4of9 rename to assets/tensorflow/style_js/group1-shard4of9 diff --git a/tf_models/style_js/group1-shard5of9 b/assets/tensorflow/style_js/group1-shard5of9 similarity index 100% rename from tf_models/style_js/group1-shard5of9 rename to assets/tensorflow/style_js/group1-shard5of9 diff --git a/tf_models/style_js/group1-shard6of9 b/assets/tensorflow/style_js/group1-shard6of9 similarity index 100% rename from tf_models/style_js/group1-shard6of9 rename to assets/tensorflow/style_js/group1-shard6of9 diff --git a/tf_models/style_js/group1-shard7of9 b/assets/tensorflow/style_js/group1-shard7of9 similarity index 100% rename from tf_models/style_js/group1-shard7of9 rename to assets/tensorflow/style_js/group1-shard7of9 diff --git a/tf_models/style_js/group1-shard8of9 b/assets/tensorflow/style_js/group1-shard8of9 similarity index 100% rename from tf_models/style_js/group1-shard8of9 rename to assets/tensorflow/style_js/group1-shard8of9 diff --git a/tf_models/style_js/group1-shard9of9 b/assets/tensorflow/style_js/group1-shard9of9 similarity index 100% rename from tf_models/style_js/group1-shard9of9 rename to assets/tensorflow/style_js/group1-shard9of9 diff --git a/tf_models/style_js/model.json b/assets/tensorflow/style_js/model.json similarity index 100% rename from tf_models/style_js/model.json rename to assets/tensorflow/style_js/model.json diff --git a/tf_models/transformer_js/group1-shard1of2 b/assets/tensorflow/transformer_js/group1-shard1of2 similarity index 100% rename from tf_models/transformer_js/group1-shard1of2 rename to assets/tensorflow/transformer_js/group1-shard1of2 diff --git a/tf_models/transformer_js/group1-shard2of2 b/assets/tensorflow/transformer_js/group1-shard2of2 similarity index 100% rename from tf_models/transformer_js/group1-shard2of2 rename to assets/tensorflow/transformer_js/group1-shard2of2 diff --git a/tf_models/transformer_js/model.json b/assets/tensorflow/transformer_js/model.json similarity index 100% rename from tf_models/transformer_js/model.json rename to assets/tensorflow/transformer_js/model.json diff --git a/structures/Tensorflow.js b/structures/Tensorflow.js index 47238b24..36e21a82 100644 --- a/structures/Tensorflow.js +++ b/structures/Tensorflow.js @@ -29,7 +29,7 @@ module.exports = class Tensorflow { async loadStyleModel() { const model = await tf.loadGraphModel( - url.pathToFileURL(path.join(__dirname, '..', 'tf_models', 'style_js', 'model.json')).href + url.pathToFileURL(path.join(__dirname, '..', 'assets', 'tensorflow', 'style_js', 'model.json')).href ); this.styleModel = model; return this.styleModel; @@ -37,7 +37,7 @@ module.exports = class Tensorflow { async loadTransformerModel() { const model = await tf.loadGraphModel( - url.pathToFileURL(path.join(__dirname, '..', 'tf_models', 'transformer_js', 'model.json')).href + url.pathToFileURL(path.join(__dirname, '..', 'assets', 'tensorflow', 'transformer_js', 'model.json')).href ); this.transformerModel = model; return this.transformerModel;