Update tts_models schema to track TalkNet-related details

migrations/2021-09-10-074547_alter_tables_talknet_metadata/down.sql

@@ -0,0 +1,15 @@
+-- noinspection SqlDialectInspectionForFile
+-- noinspection SqlNoDataSourceInspectionForFile
+-- noinspection SqlResolveForFile
+
+ALTER TABLE tts_models
+DROP COLUMN private_bucket_object_is_archive;
+
+ALTER TABLE tts_models
+DROP COLUMN has_self_contained_vocoder;
+
+ALTER TABLE tts_models
+DROP COLUMN has_self_contained_duration_model;
+
+ALTER TABLE tts_models
+DROP COLUMN has_self_contained_pitch_model;

migrations/2021-09-10-074547_alter_tables_talknet_metadata/up.sql

@@ -0,0 +1,19 @@
+-- noinspection SqlDialectInspectionForFile
+-- noinspection SqlNoDataSourceInspectionForFile
+-- noinspection SqlResolveForFile
+
+ALTER TABLE tts_models
+ADD COLUMN private_bucket_object_is_archive BOOLEAN NOT NULL DEFAULT FALSE
+AFTER private_bucket_object_name;
+
+ALTER TABLE tts_models
+ADD COLUMN has_self_contained_vocoder BOOLEAN NOT NULL DEFAULT FALSE
+AFTER tts_model_type;
+
+ALTER TABLE tts_models
+ADD COLUMN has_self_contained_duration_model BOOLEAN NOT NULL DEFAULT FALSE
+AFTER has_self_contained_vocoder;
+
+ALTER TABLE tts_models
+ADD COLUMN has_self_contained_pitch_model BOOLEAN NOT NULL DEFAULT FALSE
+AFTER has_self_contained_duration_model;

migrations_squashed/tts_tables/tts_models.sql

@@ -22,6 +22,18 @@ CREATE TABLE tts_models (
       'talknet'
   ) NOT NULL DEFAULT 'not-set',
 
+  -- For models distributed as archives/bundles (eg. TalkNet), we can pack
+  -- in a vocoder. Note that this vocoder does not have to be used.
+  has_self_contained_vocoder BOOLEAN NOT NULL DEFAULT FALSE,
+
+  -- For models distributed as archives/bundles (eg. TalkNet), we can pack
+  -- in a duration model.
+  has_self_contained_duration_model BOOLEAN NOT NULL DEFAULT FALSE,
+
+  -- For models distributed as archives/bundles (eg. TalkNet), we can pack
+  -- in a pitch model.
+  has_self_contained_pitch_model BOOLEAN NOT NULL DEFAULT FALSE,
+
   -- NB: DO NOT CHANGE ORDER; APPEND ONLY!
   -- How text should be handled.
   text_preprocessing_algorithm ENUM(
@@ -30,6 +42,7 @@ CREATE TABLE tts_models (
   ) NOT NULL DEFAULT 'basic',
 
   -- If set, use a different pretrained vocoder.
+  -- If not set, use the website default. (Currently HifiGan-SS)
   maybe_default_pretrained_vocoder VARCHAR(64) DEFAULT NULL,
 
   -- Optional Pointer to a newer version of the voice
@@ -109,6 +122,9 @@ CREATE TABLE tts_models (
   -- The "full url" version of the path
   private_bucket_object_name VARCHAR(255) NOT NULL,
 
+  -- Whether the file is a zip archive (rather than a model file).
+  private_bucket_object_is_archive BOOLEAN NOT NULL DEFAULT FALSE,
+
   -- ========== RATING AND RANKING ==========
 
   -- Calculated average, on a scale of 0-100

src/database/helpers/enums.rs

@@ -26,6 +26,8 @@ pub enum TtsModelType {
   GlowTts,
   /// glowtts-vocodes
   GlowTts_Vocodes,
+  /// talknet
+  Talknet,
 }
 
 #[derive(Deserialize)]