This commit is contained in:
Brandon Thomas committed 2024-01-23 12:50:06 -05:00
1 parent 2c8e9ff823
commit af8fb02078
2 files changed
+11 -7

No files matched your search

@@ -21,6 +21,7 @@ const PAGE_SIZE: u64 = 10;
pub async fn migrate_tts_to_weights(deps: &Deps) -> AnyhowResult<()> {
let mut cursor = 0;
let mut counter = 0;
loop {
info!("Querying {PAGE_SIZE} models at cursor = {cursor}");
@@ -34,15 +35,16 @@ pub async fn migrate_tts_to_weights(deps: &Deps) -> AnyhowResult<()> {
}
for result in results.iter() {
println!("\n\nmigrating over result {:?} : {:?} ...", result.token, result);
println!("\n\nmigrating over (counter: {counter}, id: {}, token: {:?}) : {:?} ...", result.id, result.token, result);
copy_model_record_and_bucket_files(result, &deps).await?;
counter += 1;
}
if let Some(last_id) = results.last().map(|result| result.id) {
cursor = last_id as u64;
}
thread::sleep(Duration::from_secs(2));
thread::sleep(Duration::from_millis(500));
}
Ok(())
@@ -29,9 +29,11 @@ pub struct WholeTtsModelRecord {
pub text_preprocessing_algorithm: Option<String>,
// Weird mel multiplication hack
pub use_default_mel_multiply_factor: bool,
pub maybe_custom_mel_multiply_factor: Option<f64>,
pub maybe_custom_mel_multiply_factor: Option<f64>, // 10
// Vocoder
pub maybe_default_pretrained_vocoder: Option<String>,
pub maybe_custom_vocoder_token: Option<VocoderModelToken>,
@@ -44,7 +46,7 @@ pub struct WholeTtsModelRecord {
pub voice_name: Option<String>,
pub actor_name: Option<String>,
pub characteristic: Option<String>,
pub characteristic: Option<String>, // 20
pub version_string: Option<String>,
pub is_front_page_featured: bool,
@@ -59,7 +61,7 @@ pub struct WholeTtsModelRecord {
pub creator_user_token: UserToken,
pub creator_ip_address_creation: String,
pub creator_ip_address_last_update: String,
pub creator_ip_address_last_update: String, // 30
pub creator_set_visibility: Visibility,
@@ -74,7 +76,7 @@ pub struct WholeTtsModelRecord {
pub is_public_listing_approved: Option<bool>,
pub is_locked_from_user_modification: bool,
pub is_locked_from_use: bool,
pub is_locked_from_use: bool, // 40
pub maybe_mod_comments: Option<String>,
pub maybe_mod_user_token: Option<UserToken>,
@@ -87,7 +89,7 @@ pub struct WholeTtsModelRecord {
pub updated_at: DateTime<Utc>,
pub user_deleted_at: Option<DateTime<Utc>>,
pub mod_deleted_at: Option<DateTime<Utc>>,
pub mod_deleted_at: Option<DateTime<Utc>>, // 48 fields total
}
pub async fn list_whole_tts_models_using_cursor(