Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,53 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_117_3f7a73b6eb939439853b.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_117_51c4feb7b25049501e37.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_117_6d818be44006fe311b25.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_117_7633a61e6ee80a024fb2.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_117_855fad46eee4554f15a7.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_117_dd8bdcb385062c83d9db.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_117_ea76ce7bbfab47610005.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_195_2abe63a219dd9980a311.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_195_69610571800405c8afc4.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_195_6cac4adea8fad0e75539.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_195_830d908cf01e69e53001.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_195_83726d791dfb671cadab.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_195_c0a3f3d7cd9b5e6f1ff3.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_195_c762fd4cdeb8dd603c93.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_234_07ddb3b15d19fc0d28b8.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_234_081798576741dcab4ed7.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_234_1c742d222224ecad309d.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_234_823c293a773a7747aae9.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_234_b9550b7eaa369565c70f.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_234_c4854e1ae7d92c3a58d5.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_234_e8462d5ee69db7355157.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_273_1b11908a3b16b03f6f28.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_273_623fa20cd98668542c5a.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_273_7e85e65244c43352d3ff.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_273_85b843d1a8a6ba22faee.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_273_89094caf461fb7b67ec7.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_273_8f6478a9918ad2b1ae55.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_273_9f4846c46a52d5e23c8d.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_312_01c26787b801562a3789.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_312_0ceabcd2840ac8898fe1.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_312_337071167ae9bad07b71.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_312_43fb61bde32a4b6742d4.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_312_47c9ee8dc54086885fe9.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_312_ac7b82097e1e55d1368c.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_312_f460b32c3554b8680b92.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_39_6481b4598446a5223728.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_39_65c5863263ecc0063e7c.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_39_75f77b4060ad41f6cd28.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_39_91ec273933f07274531b.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_39_a92c59803d84d120f9c8.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_39_e334e92b9d2007210460.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_39_e93fbf8d0314875d9a8a.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_78_2a1aa9c4aa9b28aeff3d.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_78_31d4a3506990aa644012.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_78_6f247990e446903b7c69.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_78_ccbbfb2e4ad764ed2d56.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_78_e0a9392514f4f0b15b68.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_78_e8936f475da43789ec82.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/files/media/audio/Speech[[:space:]]samples/eval_78_ff389c0afad3d276441d.wav filter=lfs diff=lfs merge=lfs -text
+wandb/run-20240513_205708-e3sq5zz5/run-e3sq5zz5.wandb filter=lfs diff=lfs merge=lfs -text

checkpoint-144-epoch-3/optimizer.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5e728938a020748172a043132006766f4a9aea154dfa221b5849c69bc2c026c4
+size 3652763351

checkpoint-144-epoch-3/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ab339eb359286de4937c3ac9a9a0ae12209842b4166eba2d68fe723a6a438aa2
+size 2588462170

checkpoint-144-epoch-3/random_states_0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:76d33f5216003698fa81aceb24693ae6ddf73c08c289065e1b1b011a817dd1f9
+size 14408

checkpoint-144-epoch-3/scheduler.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b79932bdc8b3cbc9e9b6af8b11344bc95b9d62b0a651d065e4da34269040e929
+size 1000

checkpoint-216-epoch-5/optimizer.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:97cc912a330eb8c6ac1412de5c46e2b9733351a81bfba0066ce67ba7d9fdd9e5
+size 3652763351

checkpoint-216-epoch-5/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9d544b282e7baef80c090fb270560cee9f423f25d2b57600c847d432dcd914ef
+size 2588462170

checkpoint-216-epoch-5/random_states_0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c388279c01ff0f2d04a22cb8fd6e5570c6cb372c2116007d42c83b24ef684a3b
+size 14344

checkpoint-216-epoch-5/scheduler.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:36f275707c92f62d3c3275afc2212e2c6176cdd14a425be9eb481c26b0a5ca7c
+size 1000

checkpoint-288-epoch-7/optimizer.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:41cdea0eced4347c246a0e029cf82f4eb4bdc22ead7d900a856ec6cb9b7a898d
+size 3652763351

checkpoint-288-epoch-7/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3fe89936ea1e9a28873323963f1143fb97b5a56c9fdbd0d8ec3a168f05be5c82
+size 2588462170

checkpoint-288-epoch-7/random_states_0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:820ccb9181b74dbaba665aae0c5a53a3b77169c67cb021111e89f95714d28e6e
+size 14344

checkpoint-288-epoch-7/scheduler.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:110faff40ae64c1ebce9c93b18f5c46479bb06243d71a1cb2b42895cff384963
+size 1000

checkpoint-312-epoch-7/optimizer.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2e0e25761980181284316f2cb1aac59524475d4fbc6f9c69c3dab6034f16ceb2
+size 3652763351

checkpoint-312-epoch-7/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a66f1ca1f6eb925ab73c9aac0f1238b913c05fb45f3c5efee7cbf04a92be5959
+size 2588462170

checkpoint-312-epoch-7/random_states_0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0fc40577c9284fcb8945a3ced5c05d62aa1f031cb88b316b2d9d467872270199
+size 14408

checkpoint-312-epoch-7/scheduler.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9b2d636bd7cf005a271ad3d3b8a6af3546951cd0ccda822994dc2edbed5b1f2f
+size 1000

checkpoint-72-epoch-1/optimizer.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5d6d2b1a649dd2c1bd454f3f83513e829024d8f3b7a439ec7cdb6682f23cfc6c
+size 3652763351

checkpoint-72-epoch-1/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bcbd40f4a0118dd67ad43790695b2692d6743c50409429188a60b6aa5a5dd523
+size 2588462170

checkpoint-72-epoch-1/random_states_0.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9ac1e338ab3a6a60c25d38c4e95428abf67cf039a45489509b3e03c86383ad2a
+size 14408

checkpoint-72-epoch-1/scheduler.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:de20d262a5716e12bcc1621b1656e3c6356daf72b9e347d23c6b0423a694e28d
+size 1000

config.json

@@ -0,0 +1,276 @@
+{
+  "_name_or_path": "parler-tts/parler_tts_mini_v0.1",
+  "architectures": [
+    "ParlerTTSForConditionalGeneration"
+  ],
+  "audio_encoder": {
+    "_name_or_path": "ylacombe/dac_44khZ_8kbps",
+    "add_cross_attention": false,
+    "architectures": [
+      "DACModel"
+    ],
+    "bad_words_ids": null,
+    "begin_suppress_tokens": null,
+    "bos_token_id": null,
+    "chunk_size_feed_forward": 0,
+    "codebook_size": 1024,
+    "cross_attention_hidden_size": null,
+    "decoder_start_token_id": null,
+    "diversity_penalty": 0.0,
+    "do_sample": false,
+    "early_stopping": false,
+    "encoder_no_repeat_ngram_size": 0,
+    "eos_token_id": null,
+    "exponential_decay_length_penalty": null,
+    "finetuning_task": null,
+    "forced_bos_token_id": null,
+    "forced_eos_token_id": null,
+    "frame_rate": 86,
+    "id2label": {
+      "0": "LABEL_0",
+      "1": "LABEL_1"
+    },
+    "is_decoder": false,
+    "is_encoder_decoder": false,
+    "label2id": {
+      "LABEL_0": 0,
+      "LABEL_1": 1
+    },
+    "latent_dim": 1024,
+    "length_penalty": 1.0,
+    "max_length": 20,
+    "min_length": 0,
+    "model_bitrate": 8,
+    "model_type": "dac",
+    "no_repeat_ngram_size": 0,
+    "num_beam_groups": 1,
+    "num_beams": 1,
+    "num_codebooks": 9,
+    "num_return_sequences": 1,
+    "output_attentions": false,
+    "output_hidden_states": false,
+    "output_scores": false,
+    "pad_token_id": null,
+    "prefix": null,
+    "problem_type": null,
+    "pruned_heads": {},
+    "remove_invalid_values": false,
+    "repetition_penalty": 1.0,
+    "return_dict": true,
+    "return_dict_in_generate": false,
+    "sampling_rate": 44100,
+    "sep_token_id": null,
+    "suppress_tokens": null,
+    "task_specific_params": null,
+    "temperature": 1.0,
+    "tf_legacy_loss": false,
+    "tie_encoder_decoder": false,
+    "tie_word_embeddings": true,
+    "tokenizer_class": null,
+    "top_k": 50,
+    "top_p": 1.0,
+    "torch_dtype": "float32",
+    "torchscript": false,
+    "typical_p": 1.0,
+    "use_bfloat16": false
+  },
+  "decoder": {
+    "_name_or_path": "/fsx/yoach/tmp/artefacts/decoder_400M/",
+    "activation_dropout": 0.0,
+    "activation_function": "gelu",
+    "add_cross_attention": true,
+    "architectures": [
+      "ParlerTTSForCausalLM"
+    ],
+    "attention_dropout": 0.0,
+    "bad_words_ids": null,
+    "begin_suppress_tokens": null,
+    "bos_token_id": 1025,
+    "chunk_size_feed_forward": 0,
+    "cross_attention_hidden_size": null,
+    "decoder_start_token_id": null,
+    "diversity_penalty": 0.0,
+    "do_sample": false,
+    "dropout": 0.1,
+    "early_stopping": false,
+    "encoder_no_repeat_ngram_size": 0,
+    "eos_token_id": 1024,
+    "exponential_decay_length_penalty": null,
+    "ffn_dim": 4096,
+    "finetuning_task": null,
+    "forced_bos_token_id": null,
+    "forced_eos_token_id": null,
+    "hidden_size": 1024,
+    "id2label": {
+      "0": "LABEL_0",
+      "1": "LABEL_1"
+    },
+    "initializer_factor": 0.02,
+    "is_decoder": true,
+    "is_encoder_decoder": false,
+    "label2id": {
+      "LABEL_0": 0,
+      "LABEL_1": 1
+    },
+    "layerdrop": 0.0,
+    "length_penalty": 1.0,
+    "max_length": 20,
+    "max_position_embeddings": 4096,
+    "min_length": 0,
+    "model_type": "parler_tts_decoder",
+    "no_repeat_ngram_size": 0,
+    "num_attention_heads": 16,
+    "num_beam_groups": 1,
+    "num_beams": 1,
+    "num_codebooks": 9,
+    "num_hidden_layers": 24,
+    "num_return_sequences": 1,
+    "output_attentions": false,
+    "output_hidden_states": false,
+    "output_scores": false,
+    "pad_token_id": 1024,
+    "prefix": null,
+    "problem_type": null,
+    "pruned_heads": {},
+    "remove_invalid_values": false,
+    "repetition_penalty": 1.0,
+    "return_dict": true,
+    "return_dict_in_generate": false,
+    "scale_embedding": false,
+    "sep_token_id": null,
+    "suppress_tokens": null,
+    "task_specific_params": null,
+    "temperature": 1.0,
+    "tf_legacy_loss": false,
+    "tie_encoder_decoder": false,
+    "tie_word_embeddings": false,
+    "tokenizer_class": null,
+    "top_k": 50,
+    "top_p": 1.0,
+    "torch_dtype": "float32",
+    "torchscript": false,
+    "typical_p": 1.0,
+    "use_bfloat16": false,
+    "use_cache": true,
+    "vocab_size": 1088
+  },
+  "decoder_start_token_id": 1025,
+  "is_encoder_decoder": true,
+  "model_type": "parler_tts",
+  "pad_token_id": 1024,
+  "text_encoder": {
+    "_name_or_path": "google/flan-t5-base",
+    "add_cross_attention": false,
+    "architectures": [
+      "T5ForConditionalGeneration"
+    ],
+    "bad_words_ids": null,
+    "begin_suppress_tokens": null,
+    "bos_token_id": null,
+    "chunk_size_feed_forward": 0,
+    "classifier_dropout": 0.0,
+    "cross_attention_hidden_size": null,
+    "d_ff": 2048,
+    "d_kv": 64,
+    "d_model": 768,
+    "decoder_start_token_id": 0,
+    "dense_act_fn": "gelu_new",
+    "diversity_penalty": 0.0,
+    "do_sample": false,
+    "dropout_rate": 0.1,
+    "early_stopping": false,
+    "encoder_no_repeat_ngram_size": 0,
+    "eos_token_id": 1,
+    "exponential_decay_length_penalty": null,
+    "feed_forward_proj": "gated-gelu",
+    "finetuning_task": null,
+    "forced_bos_token_id": null,
+    "forced_eos_token_id": null,
+    "id2label": {
+      "0": "LABEL_0",
+      "1": "LABEL_1"
+    },
+    "initializer_factor": 1.0,
+    "is_decoder": false,
+    "is_encoder_decoder": true,
+    "is_gated_act": true,
+    "label2id": {
+      "LABEL_0": 0,
+      "LABEL_1": 1
+    },
+    "layer_norm_epsilon": 1e-06,
+    "length_penalty": 1.0,
+    "max_length": 20,
+    "min_length": 0,
+    "model_type": "t5",
+    "n_positions": 512,
+    "no_repeat_ngram_size": 0,
+    "num_beam_groups": 1,
+    "num_beams": 1,
+    "num_decoder_layers": 12,
+    "num_heads": 12,
+    "num_layers": 12,
+    "num_return_sequences": 1,
+    "output_attentions": false,
+    "output_hidden_states": false,
+    "output_past": true,
+    "output_scores": false,
+    "pad_token_id": 0,
+    "prefix": null,
+    "problem_type": null,
+    "pruned_heads": {},
+    "relative_attention_max_distance": 128,
+    "relative_attention_num_buckets": 32,
+    "remove_invalid_values": false,
+    "repetition_penalty": 1.0,
+    "return_dict": true,
+    "return_dict_in_generate": false,
+    "sep_token_id": null,
+    "suppress_tokens": null,
+    "task_specific_params": {
+      "summarization": {
+        "early_stopping": true,
+        "length_penalty": 2.0,
+        "max_length": 200,
+        "min_length": 30,
+        "no_repeat_ngram_size": 3,
+        "num_beams": 4,
+        "prefix": "summarize: "
+      },
+      "translation_en_to_de": {
+        "early_stopping": true,
+        "max_length": 300,
+        "num_beams": 4,
+        "prefix": "translate English to German: "
+      },
+      "translation_en_to_fr": {
+        "early_stopping": true,
+        "max_length": 300,
+        "num_beams": 4,
+        "prefix": "translate English to French: "
+      },
+      "translation_en_to_ro": {
+        "early_stopping": true,
+        "max_length": 300,
+        "num_beams": 4,
+        "prefix": "translate English to Romanian: "
+      }
+    },
+    "temperature": 1.0,
+    "tf_legacy_loss": false,
+    "tie_encoder_decoder": false,
+    "tie_word_embeddings": false,
+    "tokenizer_class": null,
+    "top_k": 50,
+    "top_p": 1.0,
+    "torch_dtype": null,
+    "torchscript": false,
+    "typical_p": 1.0,
+    "use_bfloat16": false,
+    "use_cache": true,
+    "vocab_size": 32128
+  },
+  "torch_dtype": "float32",
+  "transformers_version": "4.41.0.dev0",
+  "vocab_size": 32128
+}

generation_config.json

@@ -0,0 +1,12 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 1025,
+  "decoder_start_token_id": 1025,
+  "do_sample": true,
+  "eos_token_id": 1024,
+  "guidance_scale": 1.0,
+  "max_length": 2580,
+  "min_new_tokens": 50,
+  "pad_token_id": 1024,
+  "transformers_version": "4.41.0.dev0"
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9e39490720427289907fbf94d35a0f3d3723fee73e27bd4f09de51715d4d1eb3
+size 2588215392

preprocessor_config.json

@@ -0,0 +1,10 @@
+{
+  "chunk_length_s": null,
+  "feature_extractor_type": "EncodecFeatureExtractor",
+  "feature_size": 1,
+  "overlap": null,
+  "padding_side": "right",
+  "padding_value": 0.0,
+  "return_attention_mask": true,
+  "sampling_rate": 44100
+}

run.sh

@@ -0,0 +1,52 @@
+accelerate launch ./run_parler_tts_training.py \
+    --model_name_or_path "parler-tts/parler_tts_mini_v0.1" \
+    --feature_extractor_name "parler-tts/dac_44khZ_8kbps" \
+    --description_tokenizer_name "parler-tts/parler_tts_mini_v0.1" \
+    --prompt_tokenizer_name "parler-tts/parler_tts_mini_v0.1" \
+    --report_to "wandb" \
+    --overwrite_output_dir true \
+    --train_dataset_name "sanchit-gandhi/expresso-concatenated-half-normal" \
+    --train_metadata_dataset_name "sanchit-gandhi/expresso-concatenated-half-normal-tags-mistral" \
+    --train_dataset_config_name "read" \
+    --train_split_name "train[:-35]" \
+    --eval_dataset_name "sanchit-gandhi/expresso-concatenated-half-normal" \
+    --eval_metadata_dataset_name "sanchit-gandhi/expresso-concatenated-half-normal-tags-mistral" \
+    --eval_dataset_config_name "read" \
+    --eval_split_name "train" \
+    --max_eval_samples 8 \
+    --per_device_eval_batch_size 16 \
+    --target_audio_column_name "audio" \
+    --description_column_name "text_description" \
+    --prompt_column_name "text" \
+    --max_duration_in_seconds 30.0 \
+    --min_duration_in_seconds 2.0 \
+    --max_text_length 400 \
+    --preprocessing_num_workers 2 \
+    --do_train true \
+    --num_train_epochs 8 \
+    --gradient_accumulation_steps 8 \
+    --gradient_checkpointing true \
+    --per_device_train_batch_size 16 \
+    --learning_rate 0.00008 \
+    --adam_beta1 0.9 \
+    --adam_beta2 0.99 \
+    --weight_decay 0.01 \
+    --lr_scheduler_type "cosine" \
+    --warmup_steps 250 \
+    --logging_steps 2 \
+    --freeze_text_encoder true \
+    --audio_encoder_per_device_batch_size 4 \
+    --dtype "bfloat16" \
+    --seed 456 \
+    --output_dir "./" \
+    --temporary_save_to_disk "../audio_code_tmp_concat/" \
+    --save_to_disk "../tmp_dataset_audio_concat/" \
+    --dataloader_num_workers 4 \
+    --do_eval \
+    --predict_with_generate \
+    --include_inputs_for_metrics \
+    --save_strategy "steps" \
+    --save_steps 72 \
+    --evaluation_strategy "epoch" \
+    --save_total_limit 5 \
+    --group_by_length true

run_parler_tts_training.py

@@ -0,0 +1,1763 @@
+#!/usr/bin/env python
+# coding=utf-8
+# Copyright 2024 The HuggingFace Inc. team. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+""" Train Parler-TTS using 🤗 Accelerate"""
+
+import logging
+import os
+import re
+import shutil
+import sys
+import time
+from dataclasses import dataclass, field
+from datetime import timedelta
+from pathlib import Path
+from typing import Dict, List, Optional, Set, Union
+
+import datasets
+import evaluate
+import numpy as np
+import torch
+import transformers
+from accelerate import Accelerator
+from accelerate.utils import AutocastKwargs, InitProcessGroupKwargs, TorchDynamoPlugin, set_seed
+from accelerate.utils.memory import release_memory
+from datasets import Dataset, DatasetDict, IterableDataset, concatenate_datasets, interleave_datasets, load_dataset
+from huggingface_hub import Repository, create_repo
+from multiprocess import set_start_method
+from torch.utils.data import DataLoader
+from tqdm import tqdm
+from transformers import (
+    AutoFeatureExtractor,
+    AutoModel,
+    AutoProcessor,
+    AutoTokenizer,
+    HfArgumentParser,
+    Seq2SeqTrainingArguments,
+    pipeline,
+)
+from transformers.optimization import get_scheduler
+from transformers.trainer_pt_utils import LengthGroupedSampler
+from transformers.utils import send_example_telemetry
+from wandb import Audio
+
+from parler_tts import (
+    ParlerTTSConfig,
+    ParlerTTSForConditionalGeneration,
+    build_delay_pattern_mask,
+)
+
+
+logger = logging.getLogger(__name__)
+
+
+def list_field(default=None, metadata=None):
+    return field(default_factory=lambda: default, metadata=metadata)
+
+
+_RE_CHECKPOINT = re.compile(r"^checkpoint-(\d+)-epoch-(\d+)$")
+
+
+def get_last_checkpoint(folder):
+    content = os.listdir(folder)
+    checkpoints = [
+        path
+        for path in content
+        if _RE_CHECKPOINT.search(path) is not None and os.path.isdir(os.path.join(folder, path))
+    ]
+    if len(checkpoints) == 0:
+        return
+    return os.path.join(folder, max(checkpoints, key=lambda x: int(_RE_CHECKPOINT.search(x).groups()[0])))
+
+
+def sorted_checkpoints(output_dir=None, checkpoint_prefix="checkpoint") -> List[str]:
+    """Helper function to sort saved checkpoints from oldest to newest."""
+    ordering_and_checkpoint_path = []
+
+    glob_checkpoints = [str(x) for x in Path(output_dir).glob(f"{checkpoint_prefix}-*") if os.path.isdir(x)]
+
+    for path in glob_checkpoints:
+        regex_match = re.match(f".*{checkpoint_prefix}-([0-9]+)", path)
+        if regex_match is not None and regex_match.groups() is not None:
+            ordering_and_checkpoint_path.append((int(regex_match.groups()[0]), path))
+
+    checkpoints_sorted = sorted(ordering_and_checkpoint_path)
+    checkpoints_sorted = [checkpoint[1] for checkpoint in checkpoints_sorted]
+    return checkpoints_sorted
+
+
+def rotate_checkpoints(save_total_limit=None, output_dir=None, checkpoint_prefix="checkpoint") -> None:
+    """Helper function to delete old checkpoints."""
+    if save_total_limit is None or save_total_limit <= 0:
+        return
+    # Check if we should delete older checkpoint(s)
+    checkpoints_sorted = sorted_checkpoints(output_dir=output_dir, checkpoint_prefix=checkpoint_prefix)
+    if len(checkpoints_sorted) <= save_total_limit:
+        return
+
+    number_of_checkpoints_to_delete = max(0, len(checkpoints_sorted) - save_total_limit)
+    checkpoints_to_be_deleted = checkpoints_sorted[:number_of_checkpoints_to_delete]
+    for checkpoint in checkpoints_to_be_deleted:
+        logger.info(f"Deleting older checkpoint [{checkpoint}] due to args.save_total_limit")
+        shutil.rmtree(checkpoint, ignore_errors=True)
+
+
+def log_metric(
+    accelerator,
+    metrics: Dict,
+    train_time: float,
+    step: int,
+    epoch: int,
+    learning_rate: float = None,
+    prefix: str = "train",
+):
+    """Helper function to log all training/evaluation metrics with the correct prefixes and styling."""
+    log_metrics = {}
+    for k, v in metrics.items():
+        log_metrics[f"{prefix}/{k}"] = v
+    log_metrics[f"{prefix}/time"] = train_time
+    log_metrics[f"{prefix}/epoch"] = epoch
+    if learning_rate is not None:
+        log_metrics[f"{prefix}/learning_rate"] = learning_rate
+    accelerator.log(log_metrics, step=step)
+
+
+def log_pred(
+    accelerator,
+    pred_descriptions: List[str],
+    pred_prompts: List[str],
+    transcriptions: List[str],
+    audios: List[torch.Tensor],
+    sampling_rate: int,
+    step: int,
+    prefix: str = "eval",
+    num_lines: int = 200000,
+):
+    """Helper function to log target/predicted transcriptions to weights and biases (wandb)."""
+    if accelerator.is_main_process:
+        wandb_tracker = accelerator.get_tracker("wandb")
+        # pretty name for current step: step 50000 -> step 50k
+        cur_step_pretty = f"{int(step // 1000)}k" if step > 1000 else step
+        prefix_pretty = prefix.replace("/", "-")
+
+        # convert str data to a wandb compatible format
+        str_data = [[pred_descriptions[i], pred_prompts[i], transcriptions[i]] for i in range(len(pred_descriptions))]
+        # log as a table with the appropriate headers
+        wandb_tracker.log_table(
+            table_name=f"predictions/{prefix_pretty}-step-{cur_step_pretty}",
+            columns=["Target descriptions", "Target prompts", "Predicted transcriptions"],
+            data=str_data[:num_lines],
+            step=step,
+            commit=False,
+        )
+
+        # wandb can only loads 100 audios per step
+        wandb_tracker.log(
+            {
+                f"Speech samples/{prefix}": [
+                    Audio(
+                        audio,
+                        caption=f"{pred_prompts[i]} --- DESCRIPTION: {pred_descriptions[i]}",
+                        sample_rate=sampling_rate,
+                    )
+                    for (i, audio) in enumerate(audios[: min(len(audios), 100)])
+                ]
+            },
+            step=step,
+        )
+
+
+@dataclass
+class ModelArguments:
+    """
+    Arguments pertaining to which model/config/tokenizer we are going to fine-tune from.
+    """
+
+    model_name_or_path: str = field(
+        metadata={"help": "Path to pretrained model or model identifier from huggingface.co/models"}
+    )
+    config_name: Optional[str] = field(
+        default=None, metadata={"help": "Pretrained config name or path if not the same as model_name"}
+    )
+    feature_extractor_name: Optional[str] = field(
+        default=None, metadata={"help": "Pretrained feature extractor name or path if not the same as model_name"}
+    )
+    description_tokenizer_name: Optional[str] = field(
+        default=None, metadata={"help": "Pretrained description tokenizer name or path if not the same as model_name"}
+    )
+    prompt_tokenizer_name: Optional[str] = field(
+        default=None,
+        metadata={"help": "Pretrained prompt tokenizer name or path if not the same as description_tokenizer_name"},
+    )
+    cache_dir: Optional[str] = field(
+        default=None,
+        metadata={"help": "Where to store the pretrained models downloaded from huggingface.co"},
+    )
+    use_fast_tokenizer: bool = field(
+        default=True,
+        metadata={"help": "Whether to use one of the fast tokenizer (backed by the tokenizers library) or not."},
+    )
+    model_revision: str = field(
+        default="main",
+        metadata={"help": "The specific model version to use (can be a branch name, tag name or commit id)."},
+    )
+    pad_token_id: int = field(
+        default=None,
+        metadata={"help": "If specified, change the model pad token id."},
+    )
+    decoder_start_token_id: int = field(
+        default=None,
+        metadata={"help": "If specified, change the model decoder start token id."},
+    )
+    freeze_text_encoder: bool = field(
+        default=False,
+        metadata={"help": "Whether to freeze the text encoder."},
+    )
+    do_sample: bool = field(
+        default=True,
+        metadata={"help": "Whether to do sampling or greedy decoding."},
+    )
+    temperature: float = field(
+        default=1.0,
+        metadata={"help": "Temperature if sampling."},
+    )
+    max_length: int = field(
+        default=2580,
+        metadata={"help": "Generation max length."},
+    )
+    bandwidth: float = field(
+        default=6,
+        metadata={"help": "Audio encoder bandwidth."},
+    )
+    asr_model_name_or_path: str = field(
+        default="distil-whisper/distil-large-v2",
+        metadata={
+            "help": "Used to compute WER during evaluation. Path to pretrained model or model identifier from huggingface.co/models"
+        },
+    )
+    clap_model_name_or_path: str = field(
+        default="laion/larger_clap_music_and_speech",
+        metadata={
+            "help": "Used to compute audio similarity during evaluation. Path to pretrained model or model identifier from huggingface.co/models"
+        },
+    )
+
+
+@dataclass
+class DataTrainingArguments:
+    """
+    Arguments pertaining to what data we are going to input our model for training and eval.
+
+    Using `HfArgumentParser` we can turn this class
+    into argparse arguments to be able to specify them on
+    the command line.
+    """
+
+    train_dataset_name: str = field(
+        default=None,
+        metadata={
+            "help": "The name of the training dataset to use (via the datasets library). Load and combine "
+            "multiple datasets by separating dataset ids by a '+' symbol. For example, to load and combine "
+            " librispeech and common voice, set `train_dataset_name='librispeech_asr+common_voice'`."
+        },
+    )
+    train_dataset_config_name: Optional[str] = field(
+        default=None,
+        metadata={
+            "help": "The configuration name of the training dataset to use (via the datasets library). Load and combine "
+            "multiple datasets by separating dataset configs by a '+' symbol."
+        },
+    )
+    train_split_name: str = field(
+        default="train",
+        metadata={
+            "help": ("The name of the training data set split to use (via the datasets library). Defaults to 'train'")
+        },
+    )
+    train_dataset_samples: str = field(
+        default=None,
+        metadata={
+            "help": "Number of samples in the training data. Load and combine "
+            "multiple datasets by separating dataset samples by a '+' symbol."
+        },
+    )
+    train_metadata_dataset_name: str = field(
+        default=None,
+        metadata={
+            "help": "The name of the metadata training dataset to use (via the datasets library). Load and combine "
+            "multiple datasets by separating dataset ids by a '+' symbol. For example, to load and combine "
+            " librispeech and common voice, set `train_dataset_name='librispeech_asr+common_voice'`."
+        },
+    )
+    eval_dataset_name: str = field(
+        default=None,
+        metadata={
+            "help": "The name of the evaluation dataset to use (via the datasets library). Defaults to the training dataset name if unspecified."
+        },
+    )
+    eval_dataset_config_name: Optional[str] = field(
+        default=None,
+        metadata={
+            "help": "The configuration name of the evaluation dataset to use (via the datasets library). Defaults to the training dataset config name if unspecified"
+        },
+    )
+    eval_split_name: str = field(
+        default="test",
+        metadata={
+            "help": "The name of the evaluation data set split to use (via the datasets library). Defaults to 'test'"
+        },
+    )
+    eval_metadata_dataset_name: str = field(
+        default=None,
+        metadata={
+            "help": "The name of the metadata training dataset to use (via the datasets library). Load and combine "
+            "multiple datasets by separating dataset ids by a '+' symbol. For example, to load and combine "
+            " librispeech and common voice, set `train_dataset_name='librispeech_asr+common_voice'`."
+        },
+    )
+    target_audio_column_name: str = field(
+        default="audio",
+        metadata={"help": "The name of the dataset column containing the target audio data. Defaults to 'audio'"},
+    )
+    description_column_name: str = field(
+        default=None,
+        metadata={"help": "The name of the dataset column containing the description text data. Defaults to 'None'."},
+    )
+    prompt_column_name: str = field(
+        default=None,
+        metadata={"help": "The name of the dataset column containing the prompt text data. Defaults to 'None'."},
+    )
+    overwrite_cache: bool = field(
+        default=False, metadata={"help": "Overwrite the cached preprocessed datasets or not."}
+    )
+    preprocessing_num_workers: Optional[int] = field(
+        default=None,
+        metadata={"help": "The number of processes to use for the preprocessing."},
+    )
+    max_train_samples: Optional[int] = field(
+        default=None,
+        metadata={
+            "help": (
+                "For debugging purposes or quicker training, truncate the number of training examples to this "
+                "value if set."
+            )
+        },
+    )
+    max_eval_samples: Optional[int] = field(
+        default=None,
+        metadata={
+            "help": (
+                "For debugging purposes or quicker training, truncate the number of validation examples to this "
+                "value if set."
+            )
+        },
+    )
+    max_duration_in_seconds: float = field(
+        default=35.0,
+        metadata={
+            "help": (
+                "Filter audio files that are longer than `max_duration_in_seconds` seconds to 'max_duration_in_seconds`."
+                "Also, used to set maximum audio length if `pad_to_max_length=True`."
+            )
+        },
+    )
+    min_duration_in_seconds: float = field(
+        default=0.0, metadata={"help": "Filter audio files that are shorter than `min_duration_in_seconds` seconds"}
+    )
+    max_text_length: int = field(
+        default=500, metadata={"help": "If set, max description lengths in number of characters."}
+    )
+    max_prompt_token_length: int = field(
+        default=None,
+        metadata={
+            "help": (
+                "If set, filter samples with prompts that are longer than `max_prompt_token_length` tokens."
+                "Also, used to set maximum prompt token length if `pad_to_max_length=True`."
+            )
+        },
+    )
+    max_description_token_length: int = field(
+        default=None,
+        metadata={
+            "help": (
+                "If set, filter samples with descriptions that are longer than `max_description_token_length` tokens."
+                "Also, used to set maximum desription token length if `pad_to_max_length=True`."
+            )
+        },
+    )
+    pad_to_max_length: bool = field(
+        default=False,
+        metadata={
+            "help": (
+                "If `True`, pad audio, prompt and description to a maximum length set with respectively "
+                "`max_duration_in_seconds`, `max_prompt_token_length`, `max_description_token_length`."
+            )
+        },
+    )
+    preprocessing_only: bool = field(
+        default=False,
+        metadata={
+            "help": (
+                "Whether to only do data preprocessing and skip training. This is especially useful when data"
+                " preprocessing errors out in distributed training due to timeout. In this case, one should run the"
+                " preprocessing in a non-distributed setup with `preprocessing_only=True` so that the cached datasets"
+                " can consequently be loaded in distributed training."
+                " In this training script, `save_to_disk` must be set to the path in which the dataset should be saved. "
+            )
+        },
+    )
+    token: str = field(
+        default=None,
+        metadata={
+            "help": (
+                "The token to use as HTTP bearer authorization for remote files. If not specified, will use the token "
+                "generated when running `huggingface-cli login` (stored in `~/.huggingface`)."
+            )
+        },
+    )
+    use_auth_token: bool = field(
+        default=None,
+        metadata={
+            "help": "The `use_auth_token` argument is deprecated and will be removed in v4.34. Please use `token` instead."
+        },
+    )
+    trust_remote_code: bool = field(
+        default=False,
+        metadata={
+            "help": (
+                "Whether or not to allow for custom models defined on the Hub in their own modeling files. This option "
+                "should only be set to `True` for repositories you trust and in which you have read the code, as it will "
+                "execute code present on the Hub on your local machine."
+            )
+        },
+    )
+    add_audio_samples_to_wandb: bool = field(
+        default=False,
+        metadata={"help": "If set and if `wandb` in args.report_to, will add generated audio samples to wandb logs."},
+    )
+    id_column_name: str = field(default=None, metadata={"help": "id column name."})
+    wandb_project: str = field(
+        default="parler-speech",
+        metadata={"help": "The name of the wandb project."},
+    )
+    save_to_disk: str = field(
+        default=None,
+        metadata={
+            "help": "If set, will save the dataset to this path if this is an empyt folder. If not empty, will load the datasets from it."
+        },
+    )
+    temporary_save_to_disk: str = field(default=None, metadata={"help": "Temporarily save audio labels here."})
+    pad_to_multiple_of: Optional[int] = field(
+        default=2,
+        metadata={"help": ("Pad to multiple of for tokenizers.")},
+    )
+
+
+@dataclass
+class ParlerTTSTrainingArguments(Seq2SeqTrainingArguments):
+    dtype: Optional[str] = field(
+        default="float32",
+        metadata={
+            "help": (
+                "The data type (dtype) in which to run training. One of `float32` (full-precision), "
+                "`float16` or `bfloat16` (both half-precision)."
+            )
+        },
+    )
+    audio_encoder_per_device_batch_size: int = field(
+        default=8,
+        metadata={"help": ("Specify the batch size of the audio encoding pre-processing steps.")},
+    )
+
+
+@dataclass
+class DataCollatorEncodecWithPadding:
+    """
+    Data collator that will dynamically pad the inputs received to the longest sequence in the batch or
+    to `max_length` if `max_length` is set and `padding=max_length`.
+    """
+
+    feature_extractor: AutoFeatureExtractor
+    audio_column_name: str
+    feature_extractor_input_name: Optional[str] = "input_values"
+    max_length: Optional[int] = None
+    padding: Optional[str] = "longest"
+
+    def __call__(self, features: List[Dict[str, Union[List[int], torch.Tensor]]]) -> Dict[str, torch.Tensor]:
+        # split inputs and labels since they have to be of different lengths and need
+        # different padding methods
+        audios = [feature[self.audio_column_name]["array"] for feature in features]
+        len_audio = [len(audio) for audio in audios]
+
+        batch = self.feature_extractor(audios, return_tensors="pt", padding=self.padding, max_length=self.max_length, sampling_rate=self.feature_extractor.sampling_rate)
+        batch["len_audio"] = torch.tensor(len_audio).unsqueeze(1)
+        return batch
+
+
+@dataclass
+class DataCollatorParlerTTSWithPadding:
+    """
+    Data collator that will dynamically pad the inputs received.
+    Args:
+        prompt_tokenizer (:class:`~transformers.AutoTokenizer`)
+            The prompt_tokenizer used for proccessing the data.
+        description_tokenizer (:class:`~transformers.AutoTokenizer`)
+            The description_tokenizer used for proccessing the data.
+        padding (:obj:`bool`, :obj:`str` or :class:`~transformers.tokenization_utils_base.PaddingStrategy`, `optional`, defaults to :obj:`True`):
+            Select a strategy to pad the returned sequences (according to the model's padding side and padding index)
+            among:
+            * :obj:`True` or :obj:`'longest'`: Pad to the longest sequence in the batch (or no padding if only a single
+              sequence if provided).
+            * :obj:`'max_length'`: Pad to a maximum length specified with the argument :obj:`max_length` or to the
+              maximum acceptable input length for the model if that argument is not provided.
+            * :obj:`False` or :obj:`'do_not_pad'` (default): No padding (i.e., can output a batch with sequences of
+              different lengths).
+        pad_to_multiple_of (:obj:`int`, `optional`):
+            If set will pad the sequence to a multiple of the provided value.
+            This is especially useful to enable the use of Tensor Cores on NVIDIA hardware with compute capability >=
+            7.5 (Volta).
+    """
+
+    prompt_tokenizer: AutoTokenizer
+    description_tokenizer: AutoTokenizer
+    padding: Union[bool, str] = "longest"
+    pad_to_multiple_of: Optional[int] = None
+    prompt_max_length: Optional[int] = None
+    description_max_length: Optional[int] = None
+    audio_max_length: Optional[int] = None
+
+    def __call__(self, features: List[Dict[str, Union[List[int], torch.Tensor]]]) -> Dict[str, torch.Tensor]:
+        # split inputs and labels since they have to be of different lengths and need
+        # different padding methods
+
+        labels = [torch.tensor(feature["labels"]).transpose(0, 1) for feature in features]
+        # (bsz, seq_len, num_codebooks)
+        labels = torch.nn.utils.rnn.pad_sequence(labels, batch_first=True, padding_value=-100)
+        if self.audio_max_length is not None and self.padding == "max_length":
+            labels = torch.nn.functional.pad(labels, pad=(0, 0, 0, max(self.audio_max_length - labels.shape[1], 0)))
+
+        input_ids = [{"input_ids": feature["input_ids"]} for feature in features]
+
+        input_ids = self.description_tokenizer.pad(
+            input_ids,
+            return_tensors="pt",
+            padding=self.padding,
+            pad_to_multiple_of=self.pad_to_multiple_of,
+            max_length=self.description_max_length,
+        )
+
+        batch = {"labels": labels, **input_ids}
+
+        if self.audio_max_length is not None and self.padding == "max_length":
+            # if we do torch.compile, we need to also specify the attention_mask
+            decoder_attention_mask = torch.ones(labels.shape[:2], dtype=input_ids["attention_mask"].dtype)
+            batch["decoder_attention_mask"] = decoder_attention_mask
+
+        prompt_input_ids = [{"input_ids": feature["prompt_input_ids"]} for feature in features]
+        prompt_input_ids = self.prompt_tokenizer.pad(
+            prompt_input_ids,
+            return_tensors="pt",
+            padding=self.padding,
+            pad_to_multiple_of=self.pad_to_multiple_of,
+            max_length=self.prompt_max_length,
+        )
+
+        batch["prompt_input_ids"] = prompt_input_ids["input_ids"]
+        if "attention_mask" in prompt_input_ids:
+            batch["prompt_attention_mask"] = prompt_input_ids["attention_mask"]
+
+        return batch
+
+
+def convert_dataset_str_to_list(
+    dataset_names,
+    dataset_config_names,
+    metadata_dataset_names=None,
+    splits=None,
+    dataset_samples=None,
+    default_split="train",
+):
+    if isinstance(dataset_names, str):
+        dataset_names = dataset_names.split("+")
+        dataset_config_names = dataset_config_names.split("+")
+        splits = splits.split("+") if splits is not None else None
+        dataset_samples = dataset_samples.split("+") if dataset_samples is not None else None
+        metadata_dataset_names = metadata_dataset_names.split("+") if metadata_dataset_names is not None else None
+
+    # basic checks to ensure we've got the right number of datasets/configs/splits/columns/probs
+    if len(dataset_names) != len(dataset_config_names):
+        raise ValueError(
+            f"Ensure one config is passed for each dataset, got {len(dataset_names)} datasets and"
+            f" {len(dataset_config_names)} configs."
+        )
+
+    if splits is not None and len(splits) != len(dataset_names):
+        raise ValueError(
+            f"Ensure one split is passed for each dataset, got {len(dataset_names)} datasets and {len(splits)} splits."
+        )
+
+    if metadata_dataset_names is not None and len(metadata_dataset_names) != len(dataset_names):
+        raise ValueError(
+            f"Ensure one metadata dataset is passed for each dataset, got {len(dataset_names)} datasets and {len(metadata_dataset_names)} metadata datasets."
+        )
+
+    if dataset_samples is not None:
+        if len(dataset_samples) != len(dataset_names):
+            raise ValueError(
+                f"Ensure one sample is passed for each dataset, got {len(dataset_names)} datasets and "
+                f"{len(dataset_samples)} samples."
+            )
+        dataset_samples = [float(ds_sample) for ds_sample in dataset_samples]
+    else:
+        dataset_samples = [None] * len(dataset_names)
+
+    splits = splits if splits is not None else [default_split for _ in range(len(dataset_names))]
+
+    dataset_names_dict = []
+    for i, ds_name in enumerate(dataset_names):
+        dataset_names_dict.append(
+            {
+                "name": ds_name,
+                "config": dataset_config_names[i],
+                "split": splits[i],
+                "metadata_dataset_name": metadata_dataset_names[i],
+                "samples": dataset_samples[i],
+            }
+        )
+    return dataset_names_dict
+
+
+def load_multiple_datasets(
+    accelerator: Accelerator,
+    dataset_names: Union[List, str],
+    dataset_config_names: Union[List, str],
+    metadata_dataset_names: Optional[str] = None,
+    splits: Optional[Union[List, str]] = None,
+    label_column_names: Optional[List] = None,
+    stopping_strategy: Optional[str] = "first_exhausted",
+    dataset_samples: Optional[Union[List, np.array]] = None,
+    streaming: Optional[bool] = False,
+    seed: Optional[int] = None,
+    id_column_name: Optional[str] = None,
+    columns_to_keep: Optional[Set[str]] = None,
+    prompt_column_name: Optional[str] = None,
+    sampling_rate: Optional[int] = None,
+    audio_column_name: Optional[str] = None,
+    **kwargs,
+) -> Union[Dataset, IterableDataset]:
+    dataset_names_dict = convert_dataset_str_to_list(
+        dataset_names, dataset_config_names, metadata_dataset_names, splits, label_column_names, dataset_samples
+    )
+
+    if dataset_samples is not None:
+        dataset_samples = [ds_dict["samples"] for ds_dict in dataset_names_dict]
+        probabilities = np.array(dataset_samples) / np.sum(dataset_samples)
+    else:
+        probabilities = None
+
+    all_datasets = []
+    # iterate over the datasets we want to interleave
+    for dataset_dict in tqdm(dataset_names_dict, desc="Combining datasets..."):
+        with accelerator.main_process_first():
+            dataset = load_dataset(
+                dataset_dict["name"],
+                dataset_dict["config"],
+                split=dataset_dict["split"],
+                streaming=streaming,
+                **kwargs,
+            )
+            dataset_features = dataset.features.keys()
+
+            if sampling_rate is not None and audio_column_name is not None:
+                # resample target audio
+                dataset = dataset.cast_column(audio_column_name, datasets.features.Audio(sampling_rate=sampling_rate))
+
+            metadata_dataset_name = dataset_dict["metadata_dataset_name"]
+            if metadata_dataset_name is not None:
+                logger.info(
+                    f'Merging {dataset_dict["name"]} - {dataset_dict["split"]} with {metadata_dataset_name} - {dataset_dict["split"]}'
+                )
+                metadata_dataset = load_dataset(
+                    metadata_dataset_name,
+                    dataset_dict["config"],
+                    split=dataset_dict["split"],
+                    streaming=streaming,
+                    **kwargs,
+                )
+
+                # TODO(YL): I forgot to create unique ids for MLS english.
+                # To iterate faster, I bypass the original id check and do another one. - Done once because assuming it won't change next time
+                # if dataset_dict["name"] == "parler-tts/mls_eng_10k":
+                #     def concat_ids(book_id, speaker_id, begin_time):
+                #         return {"id": f"{book_id}_{speaker_id}_{str(begin_time).replace('.', '_')}"}
+                #     dataset = dataset.map(concat_ids, input_columns=["book_id", "speaker_id", "begin_time"], num_proc=24)
+                #     metadata_dataset = metadata_dataset.map(concat_ids, input_columns=["book_id", "speaker_id", "begin_time"], num_proc=24)
+                #     metadata_dataset = metadata_dataset.rename_column(id_column_name, f"metadata_{id_column_name}")
+
+                if dataset_dict["name"] != "parler-tts/mls_eng_10k":
+                    if id_column_name is not None and id_column_name not in dataset.column_names:
+                        raise ValueError(
+                            f"id_column_name={id_column_name} but has not been found in the dataset columns"
+                            f"- one of {', '.join(list(dataset.column_names))}."
+                        )
+                    if id_column_name is not None and id_column_name not in metadata_dataset.column_names:
+                        raise ValueError(
+                            f"id_column_name={id_column_name} but has not been found in the metadata dataset columns"
+                            f"- one of {', '.join(list(metadata_dataset.column_names))}."
+                        )
+                    elif id_column_name is not None:
+                        metadata_dataset = metadata_dataset.rename_column(id_column_name, f"metadata_{id_column_name}")
+
+                metadata_columns_to_remove = set(metadata_dataset.column_names).intersection(set(dataset.column_names))
+
+                if prompt_column_name is not None:
+                    # We might have applied some transformations to the prompts (e.g  punctuation restoration)
+                    # so we make sure to remove it from the original dataset
+                    if prompt_column_name in dataset.column_names:
+                        logger.info(
+                            f"REMOVE {prompt_column_name} from dataset {dataset_dict['name']} - dataset_dict['split']"
+                        )
+                        dataset.remove_columns(prompt_column_name)
+
+                metadata_columns_to_remove = set(metadata_dataset.column_names).intersection(set(dataset.column_names))
+                metadata_dataset = metadata_dataset.remove_columns(metadata_columns_to_remove)
+
+                dataset = concatenate_datasets([dataset, metadata_dataset], axis=1)
+
+                if id_column_name is not None and dataset_dict["name"] != "parler-tts/mls_eng_10k":
+                    if (
+                        len(
+                            dataset.filter(
+                                lambda id1, id2: id1 != id2,
+                                input_columns=[id_column_name, f"metadata_{id_column_name}"],
+                            )
+                        )
+                        != 0
+                    ):
+                        raise ValueError(
+                            f"Concatenate didn't work. Some ids don't correspond on dataset {dataset_dict['name']}"
+                        )
+
+                dataset_features = dataset.features.keys()
+
+            if columns_to_keep is not None:
+                dataset = dataset.remove_columns(set(dataset_features - columns_to_keep))
+        all_datasets.append(dataset)
+
+    if len(all_datasets) == 1:
+        # we have a single dataset so just return it as is
+        return all_datasets[0]
+
+    if streaming:
+        interleaved_dataset = interleave_datasets(
+            all_datasets,
+            stopping_strategy=stopping_strategy,
+            probabilities=probabilities,
+            seed=seed,
+        )
+    else:
+        with accelerator.main_process_first():
+            interleaved_dataset = concatenate_datasets(all_datasets)
+
+    return interleaved_dataset
+
+
+def main():
+    # See all possible arguments in src/transformers/training_args.py
+    # or by passing the --help flag to this script.
+    # We now keep distinct sets of args, for a cleaner separation of concerns.
+
+    parser = HfArgumentParser((ModelArguments, DataTrainingArguments, ParlerTTSTrainingArguments))
+    if len(sys.argv) == 2 and sys.argv[1].endswith(".json"):
+        # If we pass only one argument to the script and it's the path to a json file,
+        # let's parse it to get our arguments.
+        model_args, data_args, training_args = parser.parse_json_file(json_file=os.path.abspath(sys.argv[1]))
+    else:
+        model_args, data_args, training_args = parser.parse_args_into_dataclasses()
+
+    # Sending telemetry. Tracking the example usage helps us better allocate resources to maintain them. The
+    # information sent is the one passed as arguments along with your Python/PyTorch versions.
+    send_example_telemetry("run_parler_tts", model_args, data_args)
+
+    if training_args.dtype == "float16":
+        mixed_precision = "fp16"
+    elif training_args.dtype == "bfloat16":
+        mixed_precision = "bf16"
+    else:
+        mixed_precision = "no"
+
+    if data_args.pad_to_max_length and (
+        data_args.max_duration_in_seconds is None
+        or data_args.max_prompt_token_length is None
+        or data_args.max_description_token_length is None
+    ):
+        raise ValueError(
+            "`pad_to_max_length` is `True` but one of the following parameters has not been set: `max_duration_in_seconds`, `max_prompt_token_length`, `max_description_token_length`"
+        )
+
+    padding = "max_length" if data_args.pad_to_max_length else "longest"
+
+    ####### A. Preparation
+    kwargs_handlers = [InitProcessGroupKwargs(timeout=timedelta(minutes=60))]
+    if training_args.torch_compile:
+        # TODO(YL): add more compile modes?
+        kwargs_handlers.append(TorchDynamoPlugin(backend="inductor", mode="default"))  # reduce-overhead
+
+    accelerator = Accelerator(
+        gradient_accumulation_steps=training_args.gradient_accumulation_steps,
+        mixed_precision=mixed_precision,
+        log_with=training_args.report_to,
+        project_dir=training_args.output_dir,
+        kwargs_handlers=kwargs_handlers,
+    )
+
+    accelerator.init_trackers(
+        project_name=data_args.wandb_project,
+        config={
+            "learning_rate": training_args.learning_rate,
+            "model_name_or_path": model_args.model_name_or_path,
+            "num_train_epochs": training_args.num_train_epochs,
+            "gradient_accumulation_steps": training_args.gradient_accumulation_steps,
+            "per_device_train_batch_size": training_args.per_device_train_batch_size,
+            "global_batch_size": training_args.per_device_train_batch_size * accelerator.num_processes,
+            "mixed_precision": mixed_precision,
+            "lr_scheduler_type": training_args.lr_scheduler_type,
+            "warmup_steps": training_args.warmup_steps,
+            "freeze_text_encoder": model_args.freeze_text_encoder,
+            "max_duration_in_seconds": data_args.max_duration_in_seconds,
+            "weight_decay": training_args.weight_decay,
+            "adam_beta1": training_args.adam_beta1,
+            "adam_beta2": training_args.adam_beta2,
+            "temperature": model_args.temperature,
+        },
+    )
+
+    # Detecting last checkpoint and eventually continue from last checkpoint
+    last_checkpoint = None
+    if os.path.isdir(training_args.output_dir) and training_args.do_train and not training_args.overwrite_output_dir:
+        last_checkpoint = get_last_checkpoint(training_args.output_dir)
+        if last_checkpoint is None and len(os.listdir(training_args.output_dir)) > 0:
+            raise ValueError(
+                f"Output directory ({training_args.output_dir}) already exists and is not empty. "
+                "Use --overwrite_output_dir to overcome."
+            )
+        elif last_checkpoint is not None and training_args.resume_from_checkpoint is None:
+            logger.info(
+                f"Checkpoint detected, resuming training at {last_checkpoint}. To avoid this behavior, change "
+                "the `--output_dir` or add `--overwrite_output_dir` to train from scratch."
+            )
+
+    # Setup logging
+    logging.basicConfig(
+        format="%(asctime)s - %(levelname)s - %(name)s - %(message)s",
+        datefmt="%m/%d/%Y %H:%M:%S",
+        handlers=[logging.StreamHandler(sys.stdout)],
+    )
+    logger.setLevel(logging.INFO if accelerator.is_main_process else logging.WARN)
+
+    # Log a small summary on each proces
+    logger.warning(
+        f"Process rank: {training_args.local_rank}, device: {training_args.device}, n_gpu: {training_args.n_gpu}, "
+        f"distributed training: {training_args.parallel_mode.value == 'distributed'}, 16-bits training: {training_args.fp16}"
+    )
+
+    # Set the verbosity to info of the Transformers logger (on main process only)
+    if accelerator.is_local_main_process:
+        datasets.utils.logging.set_verbosity_warning()
+        transformers.utils.logging.set_verbosity_info()
+    else:
+        datasets.utils.logging.set_verbosity_error()
+        transformers.utils.logging.set_verbosity_error()
+
+    logger.info("Training/evaluation parameters %s", training_args)
+
+    # Set seed before initializing model.
+    set_seed(training_args.seed)
+    num_workers = data_args.preprocessing_num_workers
+
+    # 1. First, lett's instantiate the feature extractor, tokenizers and model
+    # Note for distributed training, the .from_pretrained methods guarantee that only
+    # one local process can concurrently download model & vocab.
+
+    # load feature extractor
+    feature_extractor = AutoFeatureExtractor.from_pretrained(
+        model_args.feature_extractor_name or model_args.model_name_or_path,
+        cache_dir=model_args.cache_dir,
+        token=data_args.token,
+        trust_remote_code=data_args.trust_remote_code,
+    )
+    sampling_rate = feature_extractor.sampling_rate
+
+    # load prompt tokenizer
+    prompt_tokenizer = AutoTokenizer.from_pretrained(
+        model_args.prompt_tokenizer_name or model_args.description_tokenizer_name or model_args.model_name_or_path,
+        cache_dir=model_args.cache_dir,
+        token=data_args.token,
+        trust_remote_code=data_args.trust_remote_code,
+        use_fast=model_args.use_fast_tokenizer,
+        padding_side="left",  # prompt has to be padded on the left bc it's preprend to codebooks hidden states
+    )
+
+    # load description tokenizer
+    description_tokenizer = AutoTokenizer.from_pretrained(
+        model_args.description_tokenizer_name or model_args.model_name_or_path,
+        cache_dir=model_args.cache_dir,
+        token=data_args.token,
+        trust_remote_code=data_args.trust_remote_code,
+        use_fast=model_args.use_fast_tokenizer,
+    )
+
+    if model_args.use_fast_tokenizer:
+        logger.warning(
+            "Disabling fast tokenizer warning: https://github.com/huggingface/transformers/blob/main/src/transformers/tokenization_utils_base.py#L3231-L3235"
+        )
+        prompt_tokenizer.deprecation_warnings["Asking-to-pad-a-fast-tokenizer"] = True
+        description_tokenizer.deprecation_warnings["Asking-to-pad-a-fast-tokenizer"] = True
+
+    # 2. Now, let's load the dataset
+
+    if data_args.save_to_disk is not None:
+        os.makedirs(data_args.save_to_disk, exist_ok=True)
+
+    # assume that the dataset has been saved to `save_to_disk` if the latter is not empty
+    dataset_was_precomputed = len(os.listdir(data_args.save_to_disk)) > 0
+    if dataset_was_precomputed:
+        vectorized_datasets = datasets.load_from_disk(data_args.save_to_disk)
+    else:
+        raw_datasets = DatasetDict()
+
+        columns_to_keep = {
+            "target_audio_column_name": data_args.target_audio_column_name,
+            "prompt_column_name": data_args.prompt_column_name,
+        }
+        if data_args.description_column_name is not None:
+            columns_to_keep["description_column_name"] = data_args.description_column_name
+
+        if training_args.do_train:
+            raw_datasets["train"] = load_multiple_datasets(
+                accelerator,
+                data_args.train_dataset_name,
+                data_args.train_dataset_config_name,
+                metadata_dataset_names=data_args.train_metadata_dataset_name,
+                splits=data_args.train_split_name,
+                dataset_samples=data_args.train_dataset_samples,
+                seed=training_args.seed,
+                cache_dir=model_args.cache_dir,
+                num_proc=data_args.preprocessing_num_workers,
+                id_column_name=data_args.id_column_name,
+                columns_to_keep=columns_to_keep.values(),
+                prompt_column_name=data_args.prompt_column_name,
+                audio_column_name=data_args.target_audio_column_name,
+                sampling_rate=sampling_rate,
+                # streaming=data_args.streaming, TODO(SG): optionally enable streaming mode
+            )
+
+            for key in columns_to_keep:
+                if columns_to_keep[key] not in raw_datasets["train"].column_names:
+                    raise ValueError(
+                        f"--{key} '{columns_to_keep[key]}' not found in dataset '{data_args.train_dataset_name}'."
+                        f" Make sure to set `--{key}` to the correct audio column - one of"
+                        f" {', '.join(raw_datasets['train'].column_names)}."
+                    )
+
+            if data_args.max_train_samples is not None:
+                raw_datasets["train"] = raw_datasets["train"].select(range(data_args.max_train_samples))
+
+        if training_args.do_eval:
+            raw_datasets["eval"] = load_multiple_datasets(
+                accelerator,
+                data_args.eval_dataset_name if data_args.eval_dataset_name else data_args.train_dataset_name,
+                data_args.eval_dataset_config_name
+                if data_args.eval_dataset_config_name
+                else data_args.train_dataset_config_name,
+                metadata_dataset_names=data_args.eval_metadata_dataset_name,
+                splits=data_args.eval_split_name,
+                cache_dir=model_args.cache_dir,
+                num_proc=data_args.preprocessing_num_workers,
+                id_column_name=data_args.id_column_name,
+                columns_to_keep=columns_to_keep.values(),
+                prompt_column_name=data_args.prompt_column_name,
+                audio_column_name=data_args.target_audio_column_name,
+                sampling_rate=sampling_rate,
+                # streaming=data_args.streaming, TODO(SG): optionally enable streaming mode
+            )
+
+            if data_args.max_eval_samples is not None:
+                raw_datasets["eval"] = (
+                    raw_datasets["eval"].shuffle(seed=training_args.seed).select(range(data_args.max_eval_samples))
+                )
+
+    # 3. Next, let's load the config.
+    config = ParlerTTSConfig.from_pretrained(
+        model_args.model_name_or_path,
+        cache_dir=model_args.cache_dir,
+        token=data_args.token,
+        trust_remote_code=data_args.trust_remote_code,
+    )
+
+    # update pad token id and decoder_start_token_id
+    config.update(
+        {
+            "pad_token_id": model_args.pad_token_id if model_args.pad_token_id is not None else config.pad_token_id,
+            "decoder_start_token_id": (
+                model_args.decoder_start_token_id
+                if model_args.decoder_start_token_id is not None
+                else config.decoder_start_token_id
+            ),
+        }
+    )
+
+    # create model
+    model = ParlerTTSForConditionalGeneration.from_pretrained(
+        model_args.model_name_or_path,
+        cache_dir=model_args.cache_dir,
+        config=config,
+        token=data_args.token,
+        trust_remote_code=data_args.trust_remote_code,
+    )
+
+    # enable gradient checkpointing if necessary
+    if training_args.gradient_checkpointing:
+        model.gradient_checkpointing_enable()
+
+    # 4. Now we preprocess the datasets including loading the audio, resampling and normalization
+    # Thankfully, `datasets` takes care of automatically loading and resampling the audio,
+    # so that we just need to set the correct target sampling rate and normalize the input
+    # via the `feature_extractor`
+
+    # derive max & min input length for sample rate & max duration
+    sampling_rate = feature_extractor.sampling_rate
+    max_target_length = data_args.max_duration_in_seconds * sampling_rate
+    min_target_length = data_args.min_duration_in_seconds * sampling_rate
+    target_audio_column_name = data_args.target_audio_column_name
+    description_column_name = data_args.description_column_name
+    prompt_column_name = data_args.prompt_column_name
+    feature_extractor_input_name = feature_extractor.model_input_names[0]
+    audio_encoder_pad_token_id = config.decoder.pad_token_id
+    audio_encoder_eos_token_id = config.decoder.eos_token_id
+    audio_encoder_bos_token_id = model.generation_config.decoder_start_token_id
+    max_length = model.generation_config.max_length
+    num_codebooks = model.decoder.config.num_codebooks
+    bandwidth = model_args.bandwidth
+
+    # Freeze Encoders
+    model.freeze_encoders(model_args.freeze_text_encoder)
+
+    # Test all gather - used for warmout and avoiding timeout
+    test_tensor = torch.tensor([accelerator.process_index], device=accelerator.device)
+    gathered_tensor = accelerator.gather(test_tensor)
+    print("gathered_tensor", gathered_tensor)
+    accelerator.wait_for_everyone()
+
+    if not dataset_was_precomputed:
+        # Filter on text length
+        if description_column_name is not None and data_args.max_text_length is not None:
+            with accelerator.main_process_first():
+                # filter description that is shorter than max_text_length
+                raw_datasets = raw_datasets.filter(
+                    lambda x: len(x) < data_args.max_text_length,
+                    num_proc=num_workers,
+                    input_columns=[description_column_name],
+                )
+
+        # Preprocessing the dataset.
+        # We need to tokenize the texts.
+        def pass_through_processors(description, prompt):
+            batch = {}
+
+            batch["input_ids"] = description_tokenizer(description.strip())["input_ids"]
+            batch["prompt_input_ids"] = prompt_tokenizer(prompt.strip())["input_ids"]
+
+            return batch
+
+        with accelerator.main_process_first():
+            # this is a trick to avoid to rewrite the entire audio column which takes ages
+            vectorized_datasets = raw_datasets.map(
+                pass_through_processors,
+                remove_columns=next(iter(raw_datasets.values())).column_names,
+                input_columns=[description_column_name, prompt_column_name],
+                num_proc=num_workers,
+                desc="preprocess datasets",
+            )
+
+        # We use Accelerate to perform distributed inference
+        # T5 doesn't support fp16
+        autocast_kwargs = AutocastKwargs(enabled=(mixed_precision != "fp16"))
+
+        # Now we encode the audio labels with encodec.
+        ####### B. Encode audio
+
+        logger.info("*** Encode target audio with encodec ***")
+
+        # no need to prepare audio_decoder because used for inference without mixed precision
+        # see: https://huggingface.co/docs/accelerate/main/en/package_reference/accelerator#accelerate.Accelerator.prepare
+        if training_args.torch_compile:
+            audio_decoder = accelerator.prepare_model(model.audio_encoder, evaluation_mode=True)
+        else:
+            audio_decoder = model.audio_encoder
+
+        encoder_data_collator = DataCollatorEncodecWithPadding(
+            feature_extractor,
+            audio_column_name=target_audio_column_name,
+            feature_extractor_input_name=feature_extractor_input_name,
+            max_length=max_target_length,
+            padding=padding,
+        )
+
+        def apply_audio_decoder(batch):
+            len_audio = batch.pop("len_audio")
+            audio_decoder.to(batch["input_values"].device).eval()
+            with torch.no_grad():
+                labels = audio_decoder.encode(**batch, bandwidth=bandwidth)["audio_codes"]
+            output = {}
+            output["len_audio"] = len_audio
+            # (1, bsz, codebooks, seq_len) -> (bsz, seq_len, codebooks)
+            output["labels"] = labels.squeeze(0).transpose(1, 2)
+            output["ratio"] = torch.ones_like(len_audio) * labels.shape[-1] / len_audio.max()
+            return output
+
+        for split in vectorized_datasets:
+            data_loader = DataLoader(
+                raw_datasets[split],
+                batch_size=training_args.audio_encoder_per_device_batch_size,
+                collate_fn=encoder_data_collator,
+                num_workers=training_args.dataloader_num_workers,
+                pin_memory=True,
+            )
+            data_loader = accelerator.prepare(data_loader)
+
+            all_generated_labels = []
+            all_lens = []
+            for batch in tqdm(data_loader, disable=not accelerator.is_local_main_process):
+                generate_labels = apply_audio_decoder(batch)
+                generate_labels = accelerator.pad_across_processes(generate_labels, dim=1, pad_index=0)
+                generate_labels = accelerator.gather_for_metrics(generate_labels)
+
+                if accelerator.is_main_process:
+                    lab = generate_labels["labels"].cpu().transpose(1, 2).to(torch.int16)
+                    rat = generate_labels["ratio"].cpu().squeeze()
+                    lens = generate_labels["len_audio"].cpu().squeeze()
+                    lab = [l[:, : int(ratio * length)] for (l, ratio, length) in zip(lab, rat, lens)]
+
+                    all_generated_labels.extend(lab)
+                    all_lens.extend(lens)
+
+            # (1, codebooks, seq_len) where seq_len=1
+            bos_labels = torch.ones((1, num_codebooks, 1)) * audio_encoder_bos_token_id
+
+            if accelerator.is_main_process:
+                tmp_labels = Dataset.from_dict({"labels": all_generated_labels, "target_length": all_lens})
+                tmp_labels.save_to_disk(
+                    os.path.join(data_args.temporary_save_to_disk, split),
+                    num_proc=1 if split == "eval" else data_args.preprocessing_num_workers,
+                )
+            accelerator.wait_for_everyone()
+            del all_generated_labels
+
+            tmp_labels = datasets.load_from_disk(os.path.join(data_args.temporary_save_to_disk, split))
+            with accelerator.main_process_first():
+                vectorized_datasets[split] = concatenate_datasets([vectorized_datasets[split], tmp_labels], axis=1)
+
+            def postprocess_dataset(labels):
+                # (1, codebooks, seq_len)
+                labels = torch.tensor(labels).unsqueeze(0)
+                # add bos
+                labels = torch.cat([bos_labels, labels], dim=-1)
+
+                labels, delay_pattern_mask = build_delay_pattern_mask(
+                    labels,
+                    bos_token_id=audio_encoder_bos_token_id,
+                    pad_token_id=audio_encoder_eos_token_id,
+                    max_length=labels.shape[-1] + num_codebooks,
+                    num_codebooks=num_codebooks,
+                )
+
+                # the first ids of the delay pattern mask are precisely labels, we use the rest of the labels mask
+                # to take care of EOS
+                # we want labels to look like this:
+                #  - [B, a, b, E, E, E, E]
+                #  - [B, B, c, d, E, E, E]
+                #  - [B, B, B, e, f, E, E]
+                #  - [B, B, B, B, g, h, E]
+                labels = torch.where(delay_pattern_mask == -1, audio_encoder_eos_token_id, delay_pattern_mask)
+
+                # the first timestamp is associated to a row full of BOS, let's get rid of it
+                # we also remove the last timestampts (full of PAD)
+                output = {"labels": labels[:, 1:]}
+                return output
+
+            with accelerator.main_process_first():
+                vectorized_datasets[split] = vectorized_datasets[split].map(
+                    postprocess_dataset,
+                    num_proc=data_args.preprocessing_num_workers,  # this one is resource consuming if many processor.
+                    input_columns=["labels"],
+                    desc="Postprocessing labeling",
+                )
+
+        accelerator.free_memory()
+        del generate_labels, all_lens
+
+        with accelerator.main_process_first():
+            # NOTE: filtering is done at the end because in the `datasets` library, caching audio files is done after most operations
+            # caching audio files is time and disk-space consuming, so we want to avoid it at all costs, especially for large (>1Kh) audio datasets.
+            # That's also why we avoid to concat the processed datasets (vectorized_datasets) with the audio column present in raw_datasets.
+
+            def is_audio_in_length_range(length):
+                return length > min_target_length and length < max_target_length
+
+            # filter data that is shorter than min_target_length
+            vectorized_datasets = vectorized_datasets.filter(
+                is_audio_in_length_range,
+                num_proc=num_workers,
+                input_columns=["target_length"],
+            )
+
+            if description_column_name is not None and data_args.max_description_token_length is not None:
+                with accelerator.main_process_first():
+                    # filter description that is shorter than max_text_length
+                    vectorized_datasets = vectorized_datasets.filter(
+                        lambda x: len(x) < data_args.max_description_token_length,
+                        num_proc=num_workers,
+                        input_columns=["input_ids"],
+                    )
+
+            if data_args.max_prompt_token_length is not None:
+                with accelerator.main_process_first():
+                    # filter description that is shorter than max_text_length
+                    vectorized_datasets = vectorized_datasets.filter(
+                        lambda x: len(x) < data_args.max_prompt_token_length,
+                        num_proc=num_workers,
+                        input_columns=["prompt_input_ids"],
+                    )
+
+    if data_args.save_to_disk is not None and not dataset_was_precomputed:
+        if accelerator.is_main_process:
+            vectorized_datasets.save_to_disk(
+                data_args.save_to_disk,
+                num_proc=min(data_args.preprocessing_num_workers, len(vectorized_datasets["eval"]) - 1),
+            )
+        logger.info(f"Dataset saved at {data_args.save_to_disk}")
+
+    audio_max_length = None
+    if training_args.torch_compile:
+        audio_max_length = max(vectorized_datasets["train"]["target_length"])
+        with accelerator.main_process_first():
+            max_sample = vectorized_datasets["train"].filter(
+                lambda x: x == audio_max_length,
+                num_proc=num_workers,
+                input_columns=["target_length"],
+            )
+        audio_max_length = torch.tensor(max_sample[0]["labels"]).shape[1]
+
+    # for large datasets it is advised to run the preprocessing on a
+    # single machine first with ``args.preprocessing_only`` since there will mostly likely
+    # be a timeout when running the script in distributed mode.
+    # In a second step ``args.preprocessing_only`` can then be set to `False` to load the
+    # cached dataset
+    if data_args.preprocessing_only and data_args.save_to_disk is None:
+        raise ValueError(
+            "`preprocessing_only=True` but `save_to_disk` is not set. The latter should indicates where to save the dataset locally."
+        )
+    elif data_args.preprocessing_only:
+        logger.info(f"Data preprocessing finished. Files save at {data_args.save_to_disk}")
+        return
+
+    # 6. Next, we can prepare the training.
+
+    # Let's use word CLAP similary and WER metrics as our evaluation metrics,
+
+    # Define evaluation metrics during training, *i.e.* CLAP similarity
+    clap = AutoModel.from_pretrained(model_args.clap_model_name_or_path)
+    clap_processor = AutoProcessor.from_pretrained(model_args.clap_model_name_or_path)
+    metric = evaluate.load("wer")
+
+    def clap_similarity(texts, audios, device):
+        clap_inputs = clap_processor(text=texts, audios=audios, padding=True, return_tensors="pt").to(device)
+        clap.to(device)
+        with torch.no_grad():
+            text_features = clap.get_text_features(
+                clap_inputs["input_ids"], attention_mask=clap_inputs.get("attention_mask", None)
+            )
+            audio_features = clap.get_audio_features(clap_inputs["input_features"])
+
+            cosine_sim = torch.nn.functional.cosine_similarity(audio_features, text_features, dim=1, eps=1e-8)
+
+        clap.to("cpu")
+        clap_inputs.to("cpu")
+        return cosine_sim.mean().to("cpu")
+
+    def wer(prompts, audios, device):
+        asr_pipeline = pipeline(model=model_args.asr_model_name_or_path, device=device)
+        transcriptions = asr_pipeline(
+            [{"raw": audio, "sampling_rate": sampling_rate} for audio in audios],
+            batch_size=int(training_args.per_device_eval_batch_size),
+        )
+
+        word_error = 100 * metric.compute(
+            predictions=[t["text"].lower() for t in transcriptions], references=[t.lower() for t in prompts]
+        )
+
+        return word_error, [t["text"] for t in transcriptions]
+
+    eval_methods = {"clap": clap_similarity, "wer": wer}
+
+    def compute_metrics(audios, descriptions, prompts, device="cpu"):
+        input_ids = descriptions
+        texts = description_tokenizer.batch_decode(input_ids, skip_special_tokens=True)
+        prompts = prompt_tokenizer.batch_decode(prompts, skip_special_tokens=True)
+        audios = [a.cpu().numpy() for a in audios]
+        results = {"clap": eval_methods["clap"](texts, audios, device)}
+        word_error, transcriptions = eval_methods["wer"](prompts, audios, device)
+        results["wer"] = word_error
+
+        return results, texts, prompts, audios, transcriptions
+
+    # Define Training Schedule
+    # Store some constants
+    per_device_train_batch_size = int(training_args.per_device_train_batch_size)
+    train_batch_size = per_device_train_batch_size * accelerator.num_processes
+    gradient_accumulation_steps = int(training_args.gradient_accumulation_steps)
+    per_device_eval_batch_size = int(training_args.per_device_eval_batch_size)
+
+    if training_args.max_steps < 0:
+        num_epochs = int(training_args.num_train_epochs)
+        steps_per_epoch = len(vectorized_datasets["train"]) // (train_batch_size * gradient_accumulation_steps)
+        total_train_steps = steps_per_epoch * num_epochs
+    elif training_args.max_steps > 0:
+        logger.info("max_steps is given, it will override any value given in num_train_epochs")
+        total_train_steps = int(training_args.max_steps)
+        # Setting a very large number of epochs so we go as many times as necessary over the iterator.
+        num_epochs = sys.maxsize
+        steps_per_epoch = total_train_steps
+
+    if training_args.evaluation_strategy == "epoch":
+        eval_steps = steps_per_epoch
+    elif training_args.eval_steps is None:
+        logger.info(f"eval_steps is not set, evaluating at the end of each epoch")
+        eval_steps = steps_per_epoch
+    else:
+        eval_steps = training_args.eval_steps
+
+    if training_args.save_strategy == "epoch":
+        save_steps = steps_per_epoch
+    elif training_args.save_strategy == "steps":
+        save_steps = training_args.save_steps
+    else:
+        save_steps = sys.maxsize
+
+    # T5 doesn't support fp16
+    autocast_kwargs = AutocastKwargs(enabled=(mixed_precision != "fp16"))
+
+    # Define optimizer, LR scheduler, collator
+    optimizer = torch.optim.AdamW(
+        params=model.parameters(),
+        lr=training_args.learning_rate,
+        betas=(training_args.adam_beta1, training_args.adam_beta2),
+        eps=training_args.adam_epsilon,
+        weight_decay=training_args.weight_decay,
+    )
+
+    # LR scheduler gets stepped by `num_processes` each time -> account for this in warmup / total steps
+    lr_scheduler = get_scheduler(
+        name=training_args.lr_scheduler_type,
+        optimizer=optimizer,
+        num_warmup_steps=training_args.get_warmup_steps(total_train_steps) * accelerator.num_processes,
+        num_training_steps=total_train_steps * accelerator.num_processes,
+    )
+
+    # Instantiate custom data collator
+    data_collator = DataCollatorParlerTTSWithPadding(
+        prompt_tokenizer=prompt_tokenizer,
+        description_tokenizer=description_tokenizer,
+        pad_to_multiple_of=data_args.pad_to_multiple_of,
+        padding=padding,
+        prompt_max_length=data_args.max_prompt_token_length,
+        description_max_length=data_args.max_description_token_length,
+        audio_max_length=audio_max_length,
+    )
+
+    # Prepare everything with accelerate
+    model, optimizer, lr_scheduler = accelerator.prepare(model, optimizer, lr_scheduler)
+
+    logger.info("***** Running training *****")
+    logger.info(f"  Num examples = {total_train_steps * train_batch_size * gradient_accumulation_steps}")
+    logger.info("  Instantaneous batch size per device =" f" {per_device_train_batch_size}")
+    logger.info("  Gradient accumulation steps =" f" {gradient_accumulation_steps}")
+    logger.info(
+        f"  Total train batch size (w. parallel & distributed) = {train_batch_size * gradient_accumulation_steps}"
+    )
+    logger.info(f"  Total optimization steps = {total_train_steps}")
+
+    # ======================== Training ================================
+    train_time = 0
+    train_start = time.time()
+    steps_trained_progress_bar = tqdm(
+        range(total_train_steps), desc="Train steps ... ", position=0, disable=not accelerator.is_local_main_process
+    )
+    continue_training = True
+    epochs_trained = 0
+    cur_step = 0
+
+    checkpoint = None
+    if training_args.resume_from_checkpoint is not None:
+        checkpoint = training_args.resume_from_checkpoint
+    elif last_checkpoint is not None:
+        checkpoint = last_checkpoint
+
+    if accelerator.is_main_process:
+        if training_args.push_to_hub:
+            # Retrieve of infer repo_name
+            repo_name = training_args.hub_model_id
+            if repo_name is None:
+                repo_name = Path(training_args.output_dir).absolute().name
+            # Create repo and retrieve repo_id
+            repo_id = create_repo(repo_name, exist_ok=True, token=training_args.hub_token).repo_id
+            # Clone repo locally
+            repo = Repository(training_args.output_dir, clone_from=repo_id, token=training_args.hub_token)
+
+            with open(os.path.join(training_args.output_dir, ".gitignore"), "w+") as gitignore:
+                if "wandb" not in gitignore:
+                    gitignore.write("wandb\n")
+        elif training_args.output_dir is not None:
+            os.makedirs(training_args.output_dir, exist_ok=True)
+    accelerator.wait_for_everyone()
+
+    # Now save everything to be able to create a single processor later
+    # make sure all processes wait until data is saved
+    with accelerator.main_process_first():
+        # only the main process saves them
+        if accelerator.is_main_process:
+            # save feature extractor, tokenizer and config
+            if (
+                model_args.prompt_tokenizer_name is None
+                and model_args.description_tokenizer_name
+                or (model_args.prompt_tokenizer_name == model_args.description_tokenizer_name)
+            ):
+                prompt_tokenizer.save_pretrained(training_args.output_dir)
+            else:
+                logger.warning(
+                    "Prompt tokenizer ('{model_args.prompt_tokenizer_name}') and description tokenizer ('{model_args.description_tokenizer_name}') are not the same. Saving only the prompt tokenizer."
+                )
+                prompt_tokenizer.save_pretrained(training_args.output_dir)
+
+            feature_extractor.save_pretrained(training_args.output_dir)
+            config.save_pretrained(training_args.output_dir)
+
+    if checkpoint is not None:
+        accelerator.load_state(checkpoint)
+        # Find num steps and epoch from saved state string pattern
+        pattern = r"checkpoint-(\d+)-epoch-(\d+)"
+        match = re.search(pattern, checkpoint)
+        cur_step = int(match.group(1))
+        epochs_trained = int(match.group(2))
+
+        logger.info("  Continuing training from checkpoint, will skip to saved global_step")
+        logger.info(f"  Continuing training from epoch {epochs_trained}")
+        logger.info(f"  Continuing training from global step {cur_step}")
+
+        steps_trained_progress_bar.update(cur_step)
+
+        for epoch in range(0, epochs_trained):
+            vectorized_datasets["train"] = vectorized_datasets["train"].shuffle(training_args.seed)
+
+        if training_args.max_steps < 0:
+            # we know exactly the number of steps per epoch, so can skip through the required number of batches
+            resume_step = (cur_step - epochs_trained * steps_per_epoch) * gradient_accumulation_steps
+        else:
+            # Currently we don't know how many steps we've taken in the current epoch
+            # So we just shuffle the dataset one extra time and start from a fresh epoch
+            # This is "good enough" for our purposes but not fully correct
+            resume_step = None
+            vectorized_datasets["train"] = vectorized_datasets["train"].shuffle(training_args.seed)
+    else:
+        resume_step = None
+
+    gen_kwargs = {
+        "do_sample": model_args.do_sample,
+        "temperature": model_args.temperature,
+        "max_length": model_args.max_length,
+    }
+
+    # Define gradient update step fn
+    def train_step(
+        batch,
+        accelerator,
+        autocast_kwargs,
+    ):
+        model.train()
+
+        if mixed_precision == "fp16":
+            # fp16 doesn't work with T5-like models
+            with accelerator.autocast(autocast_handler=autocast_kwargs):
+                if training_args.parallel_mode.value != "distributed":
+                    encoder_outputs = model.text_encoder(
+                        input_ids=batch.get("input_ids"), attention_mask=batch.get("attention_mask", None)
+                    )
+                else:
+                    encoder_outputs = model.module.text_encoder(
+                        input_ids=batch.get("input_ids"), attention_mask=batch.get("attention_mask", None)
+                    )
+                batch["encoder_outputs"] = encoder_outputs
+
+        outputs = model(**batch)
+        # CE (data) loss
+        ce_loss = outputs.loss
+
+        metrics = {"loss": ce_loss}
+        return ce_loss, metrics
+
+    # Define eval fn
+    def eval_step(
+        batch,
+        accelerator,
+        autocast_kwargs,
+    ):
+        eval_model = model if not training_args.torch_compile else model._orig_mod
+        eval_model.eval()
+
+        if mixed_precision == "fp16":
+            # fp16 doesn't work with T5-like models
+            with accelerator.autocast(autocast_handler=autocast_kwargs):
+                with torch.no_grad():
+                    if training_args.parallel_mode.value != "distributed" or training_args.torch_compile:
+                        encoder_outputs = eval_model.text_encoder(
+                            input_ids=batch.get("input_ids"), attention_mask=batch.get("attention_mask", None)
+                        )
+                    else:
+                        encoder_outputs = eval_model.module.text_encoder(
+                            input_ids=batch.get("input_ids"), attention_mask=batch.get("attention_mask", None)
+                        )
+                batch["encoder_outputs"] = encoder_outputs
+
+        with torch.no_grad():
+            outputs = eval_model(**batch)
+        # CE (data) loss
+        ce_loss = outputs.loss
+        metrics = {"loss": ce_loss}
+        return metrics
+
+    def generate_step(batch):
+        batch.pop("decoder_attention_mask", None)
+        eval_model = accelerator.unwrap_model(model, keep_fp32_wrapper=mixed_precision != "fp16").eval()
+        if training_args.torch_compile:
+            eval_model = model._orig_mod
+
+        output_audios = eval_model.generate(**batch, **gen_kwargs)
+        output_audios = accelerator.pad_across_processes(output_audios, dim=1, pad_index=0)
+        return output_audios
+
+    for epoch in range(epochs_trained, num_epochs):
+        vectorized_datasets["train"] = vectorized_datasets["train"].shuffle(training_args.seed)
+        sampler = None
+        if training_args.group_by_length:
+            sampler = LengthGroupedSampler(train_batch_size, lengths=vectorized_datasets["train"]["target_length"])
+        train_dataloader = DataLoader(
+            vectorized_datasets["train"],
+            collate_fn=data_collator,
+            batch_size=per_device_train_batch_size,
+            sampler=sampler,
+            num_workers=training_args.dataloader_num_workers,
+            pin_memory=training_args.dataloader_pin_memory,
+        )
+        train_dataloader = accelerator.prepare(train_dataloader)
+        if hasattr(train_dataloader, "dataset") and isinstance(train_dataloader.dataset, IterableDataset):
+            train_dataloader.dataset.set_epoch(epoch)
+
+        if resume_step is not None:
+            # Skip the first N batches in the dataloader when resuming from a checkpoint
+            train_dataloader = accelerator.skip_first_batches(train_dataloader, resume_step)
+            resume_step = None
+
+        for batch in train_dataloader:
+            with accelerator.accumulate(model):
+                loss, train_metric = train_step(batch, accelerator, autocast_kwargs)
+                accelerator.backward(loss)
+                if accelerator.sync_gradients:
+                    accelerator.clip_grad_norm_(model.parameters(), training_args.max_grad_norm)
+                optimizer.step()
+                lr_scheduler.step()
+                optimizer.zero_grad()
+
+            # Check if the accelerator has performed an optimization step behind the scenes
+            if accelerator.sync_gradients:
+                steps_trained_progress_bar.update(1)
+                cur_step += 1
+
+                if cur_step % training_args.logging_steps == 0:
+                    steps_trained_progress_bar.write(
+                        f"Step... ({cur_step} / {total_train_steps} | Loss:"
+                        f" {train_metric['loss']}, Learning Rate:"
+                        f" {lr_scheduler.get_last_lr()[0]})"
+                    )
+                    log_metric(
+                        accelerator,
+                        metrics=train_metric,
+                        learning_rate=lr_scheduler.get_last_lr()[0],
+                        train_time=train_time + time.time() - train_start,
+                        step=cur_step,
+                        epoch=epoch + (cur_step - epoch * steps_per_epoch) / steps_per_epoch,
+                        prefix="train",
+                    )
+
+                # save checkpoint and weights after each save_steps and at the end of training
+                if (cur_step % save_steps == 0) or cur_step == total_train_steps:
+                    intermediate_dir = os.path.join(training_args.output_dir, f"checkpoint-{cur_step}-epoch-{epoch}")
+                    # safe_serialization=False to avoid shared tensors saving issue (TODO(YL): it's a temporary fix)
+                    # https://github.com/huggingface/transformers/issues/27293#issuecomment-1872560074
+                    accelerator.save_state(output_dir=intermediate_dir, safe_serialization=False)
+                    accelerator.wait_for_everyone()
+                    if accelerator.is_main_process:
+                        rotate_checkpoints(training_args.save_total_limit, output_dir=training_args.output_dir)
+
+                        if cur_step == total_train_steps:
+                            # un-wrap student model for save
+                            unwrapped_model = accelerator.unwrap_model(model)
+                            unwrapped_model.save_pretrained(training_args.output_dir)
+
+                        if training_args.push_to_hub:
+                            repo.push_to_hub(
+                                commit_message=f"Saving train state of step {cur_step}",
+                                blocking=False,
+                            )
+
+                if training_args.do_eval and (cur_step % eval_steps == 0 or cur_step == total_train_steps):
+                    train_time += time.time() - train_start
+                    # ======================== Evaluating ==============================
+                    eval_metrics = []
+                    eval_preds = []
+                    eval_descriptions = []
+                    eval_prompts = []
+                    eval_start = time.time()
+
+                    # release training input batch
+                    batch = release_memory(batch)
+
+                    validation_dataloader = DataLoader(
+                        vectorized_datasets["eval"],
+                        collate_fn=data_collator,
+                        batch_size=per_device_eval_batch_size,
+                        drop_last=False,
+                        num_workers=training_args.dataloader_pin_memory,
+                        pin_memory=training_args.dataloader_pin_memory,
+                    )
+                    validation_dataloader = accelerator.prepare(validation_dataloader)
+
+                    for batch in tqdm(
+                        validation_dataloader,
+                        desc="Evaluating - Inference ...",
+                        position=2,
+                        disable=not accelerator.is_local_main_process,
+                    ):
+                        # Model forward
+                        eval_metric = eval_step(batch, accelerator, autocast_kwargs)
+                        eval_metric = accelerator.gather_for_metrics(eval_metric)
+                        eval_metrics.append(eval_metric)
+
+                    if training_args.predict_with_generate:
+                        validation_dataloader = DataLoader(
+                            vectorized_datasets["eval"],
+                            collate_fn=data_collator,
+                            batch_size=per_device_eval_batch_size,
+                            drop_last=False,
+                            num_workers=training_args.dataloader_pin_memory,
+                            pin_memory=training_args.dataloader_pin_memory,
+                        )
+                        validation_dataloader = accelerator.prepare(validation_dataloader)
+                        # generation
+                        for batch in tqdm(
+                            validation_dataloader,
+                            desc="Evaluating - Generation ...",
+                            position=2,
+                            disable=not accelerator.is_local_main_process,
+                        ):
+                            generated_audios = generate_step(batch)
+                            # Gather all predictions and targets
+                            generated_audios, input_ids, prompts = accelerator.pad_across_processes(
+                                (generated_audios, batch["input_ids"], batch["prompt_input_ids"]), dim=1, pad_index=0
+                            )
+                            generated_audios, input_ids, prompts = accelerator.gather_for_metrics(
+                                (generated_audios, input_ids, prompts)
+                            )
+                            eval_preds.extend(generated_audios.to("cpu"))
+                            eval_descriptions.extend(input_ids.to("cpu"))
+                            eval_prompts.extend(prompts.to("cpu"))
+
+                    eval_time = time.time() - eval_start
+                    # normalize eval metrics
+                    eval_metrics = {
+                        key: torch.mean(torch.cat([d[key].unsqueeze(0) for d in eval_metrics]))
+                        for key in eval_metrics[0]
+                    }
+
+                    # compute metrics
+                    metrics_desc = ""
+                    if training_args.predict_with_generate:
+                        metric_values, pred_descriptions, pred_prompts, audios, transcriptions = compute_metrics(
+                            eval_preds, eval_descriptions, eval_prompts, accelerator.device
+                        )
+                        eval_metrics.update(metric_values)
+                        metrics_desc = " ".join([f"Eval {key}: {value} |" for key, value in metric_values.items()])
+                        if "wandb" in training_args.report_to:
+                            log_pred(
+                                accelerator,
+                                pred_descriptions,
+                                pred_prompts,
+                                transcriptions,
+                                audios,
+                                sampling_rate=sampling_rate,
+                                step=cur_step,
+                                prefix="eval",
+                            )
+
+                    # Print metrics and update progress bar
+                    steps_trained_progress_bar.write(
+                        f"Eval results for step ({cur_step} / {total_train_steps} | Eval Loss: {eval_metrics['loss']} |"
+                        f" {metrics_desc})"
+                    )
+
+                    log_metric(
+                        accelerator,
+                        metrics=eval_metrics,
+                        train_time=eval_time,
+                        step=cur_step,
+                        epoch=epoch + (cur_step - epoch * steps_per_epoch) / steps_per_epoch,
+                        prefix="eval",
+                    )
+
+                    # release eval batch and relax metrics
+                    eval_metrics = []
+                    eval_preds = []
+                    eval_descriptions = []
+                    eval_prompts = []
+                    batch = release_memory(batch)
+
+                    # flush the train metrics
+                    train_start = time.time()
+
+                # break condition
+                if cur_step == total_train_steps:
+                    continue_training = False
+                    break
+
+        if not continue_training:
+            break
+
+    accelerator.end_training()
+
+
+if __name__ == "__main__":
+    set_start_method("spawn")
+    main()

special_tokens_map.json

@@ -0,0 +1,125 @@
+{
+  "additional_special_tokens": [
+    "<extra_id_0>",
+    "<extra_id_1>",
+    "<extra_id_2>",
+    "<extra_id_3>",
+    "<extra_id_4>",
+    "<extra_id_5>",
+    "<extra_id_6>",
+    "<extra_id_7>",
+    "<extra_id_8>",
+    "<extra_id_9>",
+    "<extra_id_10>",
+    "<extra_id_11>",
+    "<extra_id_12>",
+    "<extra_id_13>",
+    "<extra_id_14>",
+    "<extra_id_15>",
+    "<extra_id_16>",
+    "<extra_id_17>",
+    "<extra_id_18>",
+    "<extra_id_19>",
+    "<extra_id_20>",
+    "<extra_id_21>",
+    "<extra_id_22>",
+    "<extra_id_23>",
+    "<extra_id_24>",
+    "<extra_id_25>",
+    "<extra_id_26>",
+    "<extra_id_27>",
+    "<extra_id_28>",
+    "<extra_id_29>",
+    "<extra_id_30>",
+    "<extra_id_31>",
+    "<extra_id_32>",
+    "<extra_id_33>",
+    "<extra_id_34>",
+    "<extra_id_35>",
+    "<extra_id_36>",
+    "<extra_id_37>",
+    "<extra_id_38>",
+    "<extra_id_39>",
+    "<extra_id_40>",
+    "<extra_id_41>",
+    "<extra_id_42>",
+    "<extra_id_43>",
+    "<extra_id_44>",
+    "<extra_id_45>",
+    "<extra_id_46>",
+    "<extra_id_47>",
+    "<extra_id_48>",
+    "<extra_id_49>",
+    "<extra_id_50>",
+    "<extra_id_51>",
+    "<extra_id_52>",
+    "<extra_id_53>",
+    "<extra_id_54>",
+    "<extra_id_55>",
+    "<extra_id_56>",
+    "<extra_id_57>",
+    "<extra_id_58>",
+    "<extra_id_59>",
+    "<extra_id_60>",
+    "<extra_id_61>",
+    "<extra_id_62>",
+    "<extra_id_63>",
+    "<extra_id_64>",
+    "<extra_id_65>",
+    "<extra_id_66>",
+    "<extra_id_67>",
+    "<extra_id_68>",
+    "<extra_id_69>",
+    "<extra_id_70>",
+    "<extra_id_71>",
+    "<extra_id_72>",
+    "<extra_id_73>",
+    "<extra_id_74>",
+    "<extra_id_75>",
+    "<extra_id_76>",
+    "<extra_id_77>",
+    "<extra_id_78>",
+    "<extra_id_79>",
+    "<extra_id_80>",
+    "<extra_id_81>",
+    "<extra_id_82>",
+    "<extra_id_83>",
+    "<extra_id_84>",
+    "<extra_id_85>",
+    "<extra_id_86>",
+    "<extra_id_87>",
+    "<extra_id_88>",
+    "<extra_id_89>",
+    "<extra_id_90>",
+    "<extra_id_91>",
+    "<extra_id_92>",
+    "<extra_id_93>",
+    "<extra_id_94>",
+    "<extra_id_95>",
+    "<extra_id_96>",
+    "<extra_id_97>",
+    "<extra_id_98>",
+    "<extra_id_99>"
+  ],
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<pad>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

spiece.model

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d60acb128cf7b7f2536e8f38a5b18a05535c9e14c7a355904270e15b0945ea86
+size 791656

tokenizer_config.json

@@ -0,0 +1,941 @@
+{
+  "add_prefix_space": true,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32000": {
+      "content": "<extra_id_99>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32001": {
+      "content": "<extra_id_98>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32002": {
+      "content": "<extra_id_97>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32003": {
+      "content": "<extra_id_96>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32004": {
+      "content": "<extra_id_95>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32005": {
+      "content": "<extra_id_94>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32006": {
+      "content": "<extra_id_93>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32007": {
+      "content": "<extra_id_92>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32008": {
+      "content": "<extra_id_91>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32009": {
+      "content": "<extra_id_90>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32010": {
+      "content": "<extra_id_89>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32011": {
+      "content": "<extra_id_88>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32012": {
+      "content": "<extra_id_87>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32013": {
+      "content": "<extra_id_86>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32014": {
+      "content": "<extra_id_85>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32015": {
+      "content": "<extra_id_84>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32016": {
+      "content": "<extra_id_83>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32017": {
+      "content": "<extra_id_82>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32018": {
+      "content": "<extra_id_81>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32019": {
+      "content": "<extra_id_80>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32020": {
+      "content": "<extra_id_79>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32021": {
+      "content": "<extra_id_78>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32022": {
+      "content": "<extra_id_77>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32023": {
+      "content": "<extra_id_76>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32024": {
+      "content": "<extra_id_75>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32025": {
+      "content": "<extra_id_74>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32026": {
+      "content": "<extra_id_73>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32027": {
+      "content": "<extra_id_72>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32028": {
+      "content": "<extra_id_71>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32029": {
+      "content": "<extra_id_70>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32030": {
+      "content": "<extra_id_69>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32031": {
+      "content": "<extra_id_68>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32032": {
+      "content": "<extra_id_67>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32033": {
+      "content": "<extra_id_66>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32034": {
+      "content": "<extra_id_65>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32035": {
+      "content": "<extra_id_64>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32036": {
+      "content": "<extra_id_63>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32037": {
+      "content": "<extra_id_62>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32038": {
+      "content": "<extra_id_61>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32039": {
+      "content": "<extra_id_60>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32040": {
+      "content": "<extra_id_59>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32041": {
+      "content": "<extra_id_58>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32042": {
+      "content": "<extra_id_57>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32043": {
+      "content": "<extra_id_56>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32044": {
+      "content": "<extra_id_55>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32045": {
+      "content": "<extra_id_54>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32046": {
+      "content": "<extra_id_53>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32047": {
+      "content": "<extra_id_52>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32048": {
+      "content": "<extra_id_51>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32049": {
+      "content": "<extra_id_50>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32050": {
+      "content": "<extra_id_49>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32051": {
+      "content": "<extra_id_48>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32052": {
+      "content": "<extra_id_47>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32053": {
+      "content": "<extra_id_46>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32054": {
+      "content": "<extra_id_45>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32055": {
+      "content": "<extra_id_44>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32056": {
+      "content": "<extra_id_43>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32057": {
+      "content": "<extra_id_42>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32058": {
+      "content": "<extra_id_41>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32059": {
+      "content": "<extra_id_40>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32060": {
+      "content": "<extra_id_39>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32061": {
+      "content": "<extra_id_38>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32062": {
+      "content": "<extra_id_37>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32063": {
+      "content": "<extra_id_36>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32064": {
+      "content": "<extra_id_35>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32065": {
+      "content": "<extra_id_34>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32066": {
+      "content": "<extra_id_33>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32067": {
+      "content": "<extra_id_32>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32068": {
+      "content": "<extra_id_31>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32069": {
+      "content": "<extra_id_30>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32070": {
+      "content": "<extra_id_29>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32071": {
+      "content": "<extra_id_28>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32072": {
+      "content": "<extra_id_27>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32073": {
+      "content": "<extra_id_26>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32074": {
+      "content": "<extra_id_25>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32075": {
+      "content": "<extra_id_24>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32076": {
+      "content": "<extra_id_23>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32077": {
+      "content": "<extra_id_22>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32078": {
+      "content": "<extra_id_21>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32079": {
+      "content": "<extra_id_20>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32080": {
+      "content": "<extra_id_19>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32081": {
+      "content": "<extra_id_18>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32082": {
+      "content": "<extra_id_17>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32083": {
+      "content": "<extra_id_16>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32084": {
+      "content": "<extra_id_15>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32085": {
+      "content": "<extra_id_14>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32086": {
+      "content": "<extra_id_13>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32087": {
+      "content": "<extra_id_12>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32088": {
+      "content": "<extra_id_11>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32089": {
+      "content": "<extra_id_10>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32090": {
+      "content": "<extra_id_9>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32091": {
+      "content": "<extra_id_8>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32092": {
+      "content": "<extra_id_7>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32093": {
+      "content": "<extra_id_6>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32094": {
+      "content": "<extra_id_5>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32095": {
+      "content": "<extra_id_4>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32096": {
+      "content": "<extra_id_3>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32097": {
+      "content": "<extra_id_2>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32098": {
+      "content": "<extra_id_1>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32099": {
+      "content": "<extra_id_0>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [
+    "<extra_id_0>",
+    "<extra_id_1>",
+    "<extra_id_2>",
+    "<extra_id_3>",
+    "<extra_id_4>",
+    "<extra_id_5>",
+    "<extra_id_6>",
+    "<extra_id_7>",
+    "<extra_id_8>",
+    "<extra_id_9>",
+    "<extra_id_10>",
+    "<extra_id_11>",
+    "<extra_id_12>",
+    "<extra_id_13>",
+    "<extra_id_14>",
+    "<extra_id_15>",
+    "<extra_id_16>",
+    "<extra_id_17>",
+    "<extra_id_18>",
+    "<extra_id_19>",
+    "<extra_id_20>",
+    "<extra_id_21>",
+    "<extra_id_22>",
+    "<extra_id_23>",
+    "<extra_id_24>",
+    "<extra_id_25>",
+    "<extra_id_26>",
+    "<extra_id_27>",
+    "<extra_id_28>",
+    "<extra_id_29>",
+    "<extra_id_30>",
+    "<extra_id_31>",
+    "<extra_id_32>",
+    "<extra_id_33>",
+    "<extra_id_34>",
+    "<extra_id_35>",
+    "<extra_id_36>",
+    "<extra_id_37>",
+    "<extra_id_38>",
+    "<extra_id_39>",
+    "<extra_id_40>",
+    "<extra_id_41>",
+    "<extra_id_42>",
+    "<extra_id_43>",
+    "<extra_id_44>",
+    "<extra_id_45>",
+    "<extra_id_46>",
+    "<extra_id_47>",
+    "<extra_id_48>",
+    "<extra_id_49>",
+    "<extra_id_50>",
+    "<extra_id_51>",
+    "<extra_id_52>",
+    "<extra_id_53>",
+    "<extra_id_54>",
+    "<extra_id_55>",
+    "<extra_id_56>",
+    "<extra_id_57>",
+    "<extra_id_58>",
+    "<extra_id_59>",
+    "<extra_id_60>",
+    "<extra_id_61>",
+    "<extra_id_62>",
+    "<extra_id_63>",
+    "<extra_id_64>",
+    "<extra_id_65>",
+    "<extra_id_66>",
+    "<extra_id_67>",
+    "<extra_id_68>",
+    "<extra_id_69>",
+    "<extra_id_70>",
+    "<extra_id_71>",
+    "<extra_id_72>",
+    "<extra_id_73>",
+    "<extra_id_74>",
+    "<extra_id_75>",
+    "<extra_id_76>",
+    "<extra_id_77>",
+    "<extra_id_78>",
+    "<extra_id_79>",
+    "<extra_id_80>",
+    "<extra_id_81>",
+    "<extra_id_82>",
+    "<extra_id_83>",
+    "<extra_id_84>",
+    "<extra_id_85>",
+    "<extra_id_86>",
+    "<extra_id_87>",
+    "<extra_id_88>",
+    "<extra_id_89>",
+    "<extra_id_90>",
+    "<extra_id_91>",
+    "<extra_id_92>",
+    "<extra_id_93>",
+    "<extra_id_94>",
+    "<extra_id_95>",
+    "<extra_id_96>",
+    "<extra_id_97>",
+    "<extra_id_98>",
+    "<extra_id_99>"
+  ],
+  "clean_up_tokenization_spaces": true,
+  "eos_token": "</s>",
+  "extra_ids": 100,
+  "legacy": true,
+  "model_max_length": 512,
+  "pad_token": "<pad>",
+  "padding_side": "left",
+  "sp_model_kwargs": {},
+  "tokenizer_class": "T5Tokenizer",
+  "unk_token": "<unk>"
+}

wandb/debug.log

@@ -0,0 +1,35 @@
+2024-05-13 20:57:08,363 INFO    MainThread:1264178 [wandb_setup.py:_flush():76] Current SDK version is 0.17.0
+2024-05-13 20:57:08,363 INFO    MainThread:1264178 [wandb_setup.py:_flush():76] Configure stats pid to 1264178
+2024-05-13 20:57:08,363 INFO    MainThread:1264178 [wandb_setup.py:_flush():76] Loading settings from /home/sanchit/.config/wandb/settings
+2024-05-13 20:57:08,363 INFO    MainThread:1264178 [wandb_setup.py:_flush():76] Loading settings from /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/settings
+2024-05-13 20:57:08,364 INFO    MainThread:1264178 [wandb_setup.py:_flush():76] Loading settings from environment variables: {}
+2024-05-13 20:57:08,364 INFO    MainThread:1264178 [wandb_setup.py:_flush():76] Applying setup settings: {'_disable_service': False}
+2024-05-13 20:57:08,364 INFO    MainThread:1264178 [wandb_setup.py:_flush():76] Inferring run settings from compute environment: {'program_relpath': 'run_parler_tts_training.py', 'program_abspath': '/raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/run_parler_tts_training.py', 'program': '/raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/./run_parler_tts_training.py'}
+2024-05-13 20:57:08,364 INFO    MainThread:1264178 [wandb_setup.py:_flush():76] Applying login settings: {}
+2024-05-13 20:57:08,364 INFO    MainThread:1264178 [wandb_init.py:_log_setup():520] Logging user logs to /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205708-e3sq5zz5/logs/debug.log
+2024-05-13 20:57:08,364 INFO    MainThread:1264178 [wandb_init.py:_log_setup():521] Logging internal logs to /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205708-e3sq5zz5/logs/debug-internal.log
+2024-05-13 20:57:08,364 INFO    MainThread:1264178 [wandb_init.py:init():560] calling init triggers
+2024-05-13 20:57:08,364 INFO    MainThread:1264178 [wandb_init.py:init():567] wandb.init called with sweep_config: {}
+config: {}
+2024-05-13 20:57:08,364 INFO    MainThread:1264178 [wandb_init.py:init():610] starting backend
+2024-05-13 20:57:08,364 INFO    MainThread:1264178 [wandb_init.py:init():614] setting up manager
+2024-05-13 20:57:08,367 INFO    MainThread:1264178 [backend.py:_multiprocessing_setup():105] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
+2024-05-13 20:57:08,368 INFO    MainThread:1264178 [wandb_init.py:init():622] backend started and connected
+2024-05-13 20:57:08,369 INFO    MainThread:1264178 [wandb_init.py:init():711] updated telemetry
+2024-05-13 20:57:08,373 INFO    MainThread:1264178 [wandb_init.py:init():744] communicating run to backend with 90.0 second timeout
+2024-05-13 20:57:08,727 INFO    MainThread:1264178 [wandb_run.py:_on_init():2396] communicating current version
+2024-05-13 20:57:08,782 INFO    MainThread:1264178 [wandb_run.py:_on_init():2405] got version response 
+2024-05-13 20:57:08,782 INFO    MainThread:1264178 [wandb_init.py:init():795] starting run threads in backend
+2024-05-13 20:57:11,328 INFO    MainThread:1264178 [wandb_run.py:_console_start():2374] atexit reg
+2024-05-13 20:57:11,328 INFO    MainThread:1264178 [wandb_run.py:_redirect():2229] redirect: wrap_raw
+2024-05-13 20:57:11,328 INFO    MainThread:1264178 [wandb_run.py:_redirect():2294] Wrapping output streams.
+2024-05-13 20:57:11,329 INFO    MainThread:1264178 [wandb_run.py:_redirect():2319] Redirects installed.
+2024-05-13 20:57:11,329 INFO    MainThread:1264178 [wandb_init.py:init():838] run started, returning control to user process
+2024-05-13 20:57:11,330 INFO    MainThread:1264178 [wandb_run.py:_config_callback():1376] config_cb None None {'learning_rate': 8e-05, 'model_name_or_path': 'parler-tts/parler_tts_mini_v0.1', 'num_train_epochs': 8.0, 'gradient_accumulation_steps': 8, 'per_device_train_batch_size': 16, 'global_batch_size': 16, 'mixed_precision': 'bf16', 'lr_scheduler_type': 'SchedulerType.COSINE', 'warmup_steps': 250, 'freeze_text_encoder': True, 'max_duration_in_seconds': 30.0, 'weight_decay': 0.01, 'adam_beta1': 0.9, 'adam_beta2': 0.99, 'temperature': 1.0}
+2024-05-13 21:45:31,478 INFO    MainThread:1264178 [wandb_run.py:_finish():2103] finishing run sanchit-gandhi/parler-speech/e3sq5zz5
+2024-05-13 21:45:31,478 INFO    MainThread:1264178 [wandb_run.py:_atexit_cleanup():2343] got exitcode: 0
+2024-05-13 21:45:31,478 INFO    MainThread:1264178 [wandb_run.py:_restore():2326] restore
+2024-05-13 21:45:31,478 INFO    MainThread:1264178 [wandb_run.py:_restore():2332] restore done
+2024-05-13 21:45:45,893 INFO    MainThread:1264178 [wandb_run.py:_footer_history_summary_info():3994] rendering history
+2024-05-13 21:45:45,894 INFO    MainThread:1264178 [wandb_run.py:_footer_history_summary_info():4026] rendering summary
+2024-05-13 21:45:45,897 INFO    MainThread:1264178 [wandb_run.py:_footer_sync_info():3953] logging synced files

wandb/run-20240513_204644-y7fy6vtp/files/conda-environment.yaml

@@ -0,0 +1,248 @@
+name: venv
+channels:
+  - defaults
+dependencies:
+  - _libgcc_mutex=0.1=main
+  - _openmp_mutex=5.1=1_gnu
+  - bzip2=1.0.8=h5eee18b_6
+  - ca-certificates=2024.3.11=h06a4308_0
+  - ld_impl_linux-64=2.38=h1181459_1
+  - libffi=3.4.4=h6a678d5_1
+  - libgcc-ng=11.2.0=h1234567_1
+  - libgomp=11.2.0=h1234567_1
+  - libstdcxx-ng=11.2.0=h1234567_1
+  - libuuid=1.41.5=h5eee18b_0
+  - ncurses=6.4=h6a678d5_0
+  - openssl=3.0.13=h7f8727e_1
+  - pip=24.0=py311h06a4308_0
+  - python=3.11.9=h955ad1f_0
+  - readline=8.2=h5eee18b_0
+  - setuptools=69.5.1=py311h06a4308_0
+  - sqlite=3.45.3=h5eee18b_0
+  - tk=8.6.14=h39e8969_0
+  - wheel=0.43.0=py311h06a4308_0
+  - xz=5.4.6=h5eee18b_1
+  - zlib=1.2.13=h5eee18b_1
+  - pip:
+      - absl-py==2.1.0
+      - accelerate==0.30.0
+      - aiohttp==3.9.5
+      - aiosignal==1.3.1
+      - aniso8601==9.0.1
+      - annotated-types==0.6.0
+      - anyio==4.3.0
+      - argbind==0.3.7
+      - argon2-cffi==23.1.0
+      - argon2-cffi-bindings==21.2.0
+      - arrow==1.3.0
+      - asttokens==2.4.1
+      - async-lru==2.0.4
+      - attrs==23.2.0
+      - audioread==3.0.1
+      - babel==2.15.0
+      - beautifulsoup4==4.12.3
+      - bidict==0.23.1
+      - bitsandbytes==0.43.1
+      - bleach==6.1.0
+      - certifi==2024.2.2
+      - cffi==1.16.0
+      - charset-normalizer==3.3.2
+      - click==8.1.7
+      - coloredlogs==14.0
+      - comm==0.2.2
+      - contourpy==1.2.1
+      - cycler==0.12.1
+      - datasets==2.19.1
+      - debugpy==1.8.1
+      - decorator==5.1.1
+      - defusedxml==0.7.1
+      - descript-audio-codec==1.0.0
+      - descript-audiotools==0.7.2
+      - dill==0.3.8
+      - dnspython==2.3.0
+      - docker-pycreds==0.4.0
+      - docstring-parser==0.16
+      - editdistance==0.8.1
+      - einops==0.8.0
+      - et-xmlfile==1.1.0
+      - evaluate==0.4.2
+      - eventlet==0.36.1
+      - executing==2.0.1
+      - fastjsonschema==2.19.1
+      - ffmpy==0.3.2
+      - filelock==3.14.0
+      - fire==0.6.0
+      - flask==2.2.5
+      - flask-cors==4.0.1
+      - flask-restful==0.3.10
+      - flask-socketio==5.3.6
+      - flask-talisman==1.1.0
+      - flatten-dict==0.4.2
+      - fonttools==4.51.0
+      - fqdn==1.5.1
+      - frozenlist==1.4.1
+      - fsspec==2024.3.1
+      - future==1.0.0
+      - g2p==2.0.0
+      - gitdb==4.0.11
+      - gitpython==3.1.43
+      - greenlet==3.0.3
+      - grpcio==1.63.0
+      - h11==0.14.0
+      - httpcore==1.0.5
+      - httpx==0.27.0
+      - huggingface-hub==0.23.0
+      - humanfriendly==10.0
+      - idna==3.7
+      - importlib-resources==6.4.0
+      - ipdb==0.13.13
+      - ipykernel==6.29.4
+      - ipython==8.24.0
+      - isoduration==20.11.0
+      - itsdangerous==2.2.0
+      - jedi==0.19.1
+      - jinja2==3.1.4
+      - jiwer==3.0.4
+      - joblib==1.4.2
+      - json5==0.9.25
+      - jsonpointer==2.4
+      - jsonschema==4.22.0
+      - jsonschema-specifications==2023.12.1
+      - julius==0.2.7
+      - jupyter-client==8.6.1
+      - jupyter-core==5.7.2
+      - jupyter-events==0.10.0
+      - jupyter-lsp==2.2.5
+      - jupyter-server==2.14.0
+      - jupyter-server-terminals==0.5.3
+      - jupyterlab==4.2.0
+      - jupyterlab-pygments==0.3.0
+      - jupyterlab-server==2.27.1
+      - kiwisolver==1.4.5
+      - lazy-loader==0.4
+      - librosa==0.10.2
+      - llvmlite==0.42.0
+      - markdown==3.6
+      - markdown-it-py==3.0.0
+      - markdown2==2.4.13
+      - markupsafe==2.1.5
+      - matplotlib==3.8.4
+      - matplotlib-inline==0.1.7
+      - mdurl==0.1.2
+      - mistune==3.0.2
+      - mpmath==1.3.0
+      - msgpack==1.0.8
+      - multidict==6.0.5
+      - multiprocess==0.70.16
+      - munkres==1.1.4
+      - nbclient==0.10.0
+      - nbconvert==7.16.4
+      - nbformat==5.10.4
+      - nest-asyncio==1.6.0
+      - networkx==3.3
+      - notebook-shim==0.2.4
+      - numba==0.59.1
+      - numpy==1.26.4
+      - nvidia-cublas-cu12==12.1.3.1
+      - nvidia-cuda-cupti-cu12==12.1.105
+      - nvidia-cuda-nvrtc-cu12==12.1.105
+      - nvidia-cuda-runtime-cu12==12.1.105
+      - nvidia-cudnn-cu12==8.9.2.26
+      - nvidia-cufft-cu12==11.0.2.54
+      - nvidia-curand-cu12==10.3.2.106
+      - nvidia-cusolver-cu12==11.4.5.107
+      - nvidia-cusparse-cu12==12.1.0.106
+      - nvidia-nccl-cu12==2.20.5
+      - nvidia-nvjitlink-cu12==12.4.127
+      - nvidia-nvtx-cu12==12.1.105
+      - openpyxl==3.1.2
+      - overrides==7.7.0
+      - packaging==24.0
+      - pandas==2.2.2
+      - pandocfilters==1.5.1
+      - panphon==0.20.0
+      - parler-tts==0.1
+      - parso==0.8.4
+      - pexpect==4.9.0
+      - pillow==10.3.0
+      - platformdirs==4.2.1
+      - pooch==1.8.1
+      - prometheus-client==0.20.0
+      - prompt-toolkit==3.0.43
+      - protobuf==3.19.6
+      - psutil==5.9.8
+      - ptyprocess==0.7.0
+      - pure-eval==0.2.2
+      - pyarrow==16.0.0
+      - pyarrow-hotfix==0.6
+      - pycparser==2.22
+      - pydantic==2.7.1
+      - pydantic-core==2.18.2
+      - pygments==2.18.0
+      - pyloudnorm==0.1.1
+      - pyparsing==3.1.2
+      - pystoi==0.4.1
+      - python-dateutil==2.9.0.post0
+      - python-engineio==4.9.0
+      - python-json-logger==2.0.7
+      - python-socketio==5.11.2
+      - pytz==2024.1
+      - pyyaml==6.0.1
+      - pyzmq==26.0.3
+      - randomname==0.2.1
+      - rapidfuzz==3.9.0
+      - referencing==0.35.1
+      - regex==2024.4.28
+      - requests==2.31.0
+      - rfc3339-validator==0.1.4
+      - rfc3986-validator==0.1.1
+      - rich==13.7.1
+      - rpds-py==0.18.1
+      - safetensors==0.4.3
+      - scikit-learn==1.4.2
+      - scipy==1.13.0
+      - send2trash==1.8.3
+      - sentencepiece==0.2.0
+      - sentry-sdk==2.1.1
+      - setproctitle==1.3.3
+      - simple-websocket==1.0.0
+      - six==1.16.0
+      - smmap==5.0.1
+      - sniffio==1.3.1
+      - soundfile==0.12.1
+      - soupsieve==2.5
+      - soxr==0.3.7
+      - stack-data==0.6.3
+      - sympy==1.12
+      - tensorboard==2.16.2
+      - tensorboard-data-server==0.7.2
+      - termcolor==2.4.0
+      - terminado==0.18.1
+      - text-unidecode==1.3
+      - threadpoolctl==3.5.0
+      - tinycss2==1.3.0
+      - tokenizers==0.19.1
+      - torch==2.3.0
+      - torch-stoi==0.2.1
+      - torchaudio==2.3.0
+      - tornado==6.4
+      - tqdm==4.66.4
+      - traitlets==5.14.3
+      - transformers==4.41.0.dev0
+      - triton==2.3.0
+      - types-python-dateutil==2.9.0.20240316
+      - typing-extensions==4.11.0
+      - tzdata==2024.1
+      - unicodecsv==0.14.1
+      - uri-template==1.3.0
+      - urllib3==2.2.1
+      - wandb==0.17.0
+      - wcwidth==0.2.13
+      - webcolors==1.13
+      - webencodings==0.5.1
+      - websocket-client==1.8.0
+      - werkzeug==3.0.3
+      - wsproto==1.2.0
+      - xxhash==3.4.1
+      - yarl==1.9.4
+prefix: /home/sanchit/miniconda3/envs/venv

wandb/run-20240513_204644-y7fy6vtp/files/config.yaml

@@ -0,0 +1,86 @@
+wandb_version: 1
+
+_wandb:
+  desc: null
+  value:
+    python_version: 3.11.9
+    cli_version: 0.17.0
+    framework: huggingface
+    huggingface_version: 4.41.0.dev0
+    is_jupyter_run: false
+    is_kaggle_kernel: false
+    start_time: 1715626004
+    t:
+      1:
+      - 1
+      - 5
+      - 11
+      - 49
+      - 51
+      - 53
+      - 55
+      - 71
+      - 100
+      2:
+      - 1
+      - 5
+      - 11
+      - 49
+      - 51
+      - 53
+      - 55
+      - 71
+      - 100
+      3:
+      - 23
+      4: 3.11.9
+      5: 0.17.0
+      6: 4.41.0.dev0
+      8:
+      - 5
+      13: linux-x86_64
+learning_rate:
+  desc: null
+  value: 8.0e-05
+model_name_or_path:
+  desc: null
+  value: parler-tts/parler_tts_mini_v0.1
+num_train_epochs:
+  desc: null
+  value: 8.0
+gradient_accumulation_steps:
+  desc: null
+  value: 8
+per_device_train_batch_size:
+  desc: null
+  value: 16
+global_batch_size:
+  desc: null
+  value: 16
+mixed_precision:
+  desc: null
+  value: bf16
+lr_scheduler_type:
+  desc: null
+  value: SchedulerType.COSINE
+warmup_steps:
+  desc: null
+  value: 250
+freeze_text_encoder:
+  desc: null
+  value: true
+max_duration_in_seconds:
+  desc: null
+  value: 30.0
+weight_decay:
+  desc: null
+  value: 0.01
+adam_beta1:
+  desc: null
+  value: 0.9
+adam_beta2:
+  desc: null
+  value: 0.99
+temperature:
+  desc: null
+  value: 1.0

wandb/run-20240513_204644-y7fy6vtp/files/output.log

@@ -0,0 +1,187 @@
+05/13/2024 20:46:47 - WARNING - __main__ - Process rank: 0, device: cuda:0, n_gpu: 1, distributed training: False, 16-bits training: False
+05/13/2024 20:46:47 - INFO - __main__ - Training/evaluation parameters ParlerTTSTrainingArguments(
+_n_gpu=1,
+accelerator_config={'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None},
+adafactor=False,
+adam_beta1=0.9,
+adam_beta2=0.99,
+adam_epsilon=1e-08,
+audio_encoder_per_device_batch_size=4,
+auto_find_batch_size=False,
+batch_eval_metrics=False,
+bf16=False,
+bf16_full_eval=False,
+data_seed=None,
+dataloader_drop_last=False,
+dataloader_num_workers=4,
+dataloader_persistent_workers=False,
+dataloader_pin_memory=True,
+dataloader_prefetch_factor=None,
+ddp_backend=None,
+ddp_broadcast_buffers=None,
+ddp_bucket_cap_mb=None,
+ddp_find_unused_parameters=None,
+ddp_timeout=1800,
+debug=[],
+deepspeed=None,
+disable_tqdm=False,
+dispatch_batches=None,
+do_eval=True,
+do_predict=False,
+do_train=True,
+dtype=bfloat16,
+eval_accumulation_steps=None,
+eval_delay=0,
+eval_do_concat_batches=True,
+eval_steps=None,
+eval_strategy=IntervalStrategy.EPOCH,
+evaluation_strategy=epoch,
+fp16=False,
+fp16_backend=auto,
+fp16_full_eval=False,
+fp16_opt_level=O1,
+fsdp=[],
+fsdp_config={'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False},
+fsdp_min_num_params=0,
+fsdp_transformer_layer_cls_to_wrap=None,
+full_determinism=False,
+generation_config=None,
+generation_max_length=None,
+generation_num_beams=None,
+gradient_accumulation_steps=8,
+gradient_checkpointing=True,
+gradient_checkpointing_kwargs=None,
+greater_is_better=None,
+group_by_length=True,
+half_precision_backend=auto,
+hub_always_push=False,
+hub_model_id=None,
+hub_private_repo=False,
+hub_strategy=HubStrategy.EVERY_SAVE,
+hub_token=<HUB_TOKEN>,
+ignore_data_skip=False,
+include_inputs_for_metrics=True,
+include_num_input_tokens_seen=False,
+include_tokens_per_second=False,
+jit_mode_eval=False,
+label_names=None,
+label_smoothing_factor=0.0,
+learning_rate=8e-05,
+length_column_name=length,
+load_best_model_at_end=False,
+local_rank=0,
+log_level=passive,
+log_level_replica=warning,
+log_on_each_node=True,
+logging_dir=../output_dir_training_concat/runs/May13_20-46-43_hf-dgx-01,
+logging_first_step=False,
+logging_nan_inf_filter=True,
+logging_steps=2,
+logging_strategy=IntervalStrategy.STEPS,
+lr_scheduler_kwargs={},
+lr_scheduler_type=SchedulerType.COSINE,
+max_grad_norm=1.0,
+max_steps=-1,
+metric_for_best_model=None,
+mp_parameters=,
+neftune_noise_alpha=None,
+no_cuda=False,
+num_train_epochs=8.0,
+optim=OptimizerNames.ADAMW_TORCH,
+optim_args=None,
+optim_target_modules=None,
+output_dir=../output_dir_training_concat/,
+overwrite_output_dir=True,
+past_index=-1,
+per_device_eval_batch_size=16,
+per_device_train_batch_size=16,
+predict_with_generate=True,
+prediction_loss_only=False,
+push_to_hub=False,
+push_to_hub_model_id=None,
+push_to_hub_organization=None,
+push_to_hub_token=<PUSH_TO_HUB_TOKEN>,
+ray_scope=last,
+remove_unused_columns=True,
+report_to=['wandb'],
+restore_callback_states_from_checkpoint=False,
+resume_from_checkpoint=None,
+run_name=../output_dir_training_concat/,
+save_on_each_node=False,
+save_only_model=False,
+save_safetensors=True,
+save_steps=72,
+save_strategy=IntervalStrategy.STEPS,
+save_total_limit=5,
+seed=456,
+skip_memory_metrics=True,
+sortish_sampler=False,
+split_batches=None,
+tf32=None,
+torch_compile=False,
+torch_compile_backend=None,
+torch_compile_mode=None,
+torchdynamo=None,
+tpu_metrics_debug=False,
+tpu_num_cores=None,
+use_cpu=False,
+use_ipex=False,
+use_legacy_prediction_loop=False,
+use_mps_device=False,
+warmup_ratio=0.0,
+warmup_steps=250,
+weight_decay=0.01,
+)
+05/13/2024 20:46:48 - WARNING - __main__ - Disabling fast tokenizer warning: https://github.com/huggingface/transformers/blob/main/src/transformers/tokenization_utils_base.py#L3231-L3235
+loading configuration file preprocessor_config.json from cache at /raid/.cache/huggingface/models--parler-tts--dac_44khZ_8kbps/snapshots/db52bea859d9411e0beb44a3ea923a8731ee4197/preprocessor_config.json
+Feature extractor EncodecFeatureExtractor {
+  "chunk_length_s": null,
+  "feature_extractor_type": "EncodecFeatureExtractor",
+  "feature_size": 1,
+  "overlap": null,
+  "padding_side": "right",
+  "padding_value": 0.0,
+  "return_attention_mask": true,
+  "sampling_rate": 44100
+}
+loading file spiece.model from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/spiece.model
+loading file tokenizer.json from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/tokenizer.json
+loading file added_tokens.json from cache at None
+loading file special_tokens_map.json from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/special_tokens_map.json
+loading file tokenizer_config.json from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/tokenizer_config.json
+You set `add_prefix_space`. The tokenizer needs to be converted from the slow tokenizers
+loading file spiece.model from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/spiece.model
+loading file tokenizer.json from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/tokenizer.json
+loading file added_tokens.json from cache at None
+loading file special_tokens_map.json from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/special_tokens_map.json
+loading file tokenizer_config.json from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/tokenizer_config.json
+Downloading readme: 100%|██████████████████████████████████████████████████████████████| 414/414 [00:00<00:00, 2.21MB/s]
+Downloading readme:   0%|                                                                     | 0.00/414 [00:00<?, ?B/s]
+
+
+
+
+Generating train split: 100%|███████████████████████████████████████████████| 5347/5347 [00:10<00:00, 996.42 examples/s]
+Downloading readme: 100%|██████████████████████████████████████████████████████████████| 470/470 [00:00<00:00, 1.99MB/s]
+Combining datasets...:   0%|                                                                      | 0/1 [03:44<?, ?it/s]
+Traceback (most recent call last):
+  File "/raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/./run_parler_tts_training.py", line 1763, in <module>
+    main()
+  File "/raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/./run_parler_tts_training.py", line 950, in main
+    raw_datasets["train"] = load_multiple_datasets(
+                            ^^^^^^^^^^^^^^^^^^^^^^^
+  File "/raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/./run_parler_tts_training.py", line 693, in load_multiple_datasets
+    metadata_dataset = load_dataset(
+                       ^^^^^^^^^^^^^
+  File "/home/sanchit/miniconda3/envs/venv/lib/python3.11/site-packages/datasets/load.py", line 2587, in load_dataset
+    builder_instance = load_dataset_builder(
+                       ^^^^^^^^^^^^^^^^^^^^^
+  File "/home/sanchit/miniconda3/envs/venv/lib/python3.11/site-packages/datasets/load.py", line 2296, in load_dataset_builder
+    builder_instance: DatasetBuilder = builder_cls(
+                                       ^^^^^^^^^^^^
+  File "/home/sanchit/miniconda3/envs/venv/lib/python3.11/site-packages/datasets/builder.py", line 374, in __init__
+    self.config, self.config_id = self._create_builder_config(
+                                  ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
+  File "/home/sanchit/miniconda3/envs/venv/lib/python3.11/site-packages/datasets/builder.py", line 599, in _create_builder_config
+    raise ValueError(
+ValueError: BuilderConfig 'read' not found. Available: ['default']

wandb/run-20240513_204644-y7fy6vtp/files/requirements.txt

@@ -0,0 +1,225 @@
+Babel==2.15.0
+Flask-Cors==4.0.1
+Flask-RESTful==0.3.10
+Flask-SocketIO==5.3.6
+Flask==2.2.5
+GitPython==3.1.43
+Jinja2==3.1.4
+Markdown==3.6
+MarkupSafe==2.1.5
+PyYAML==6.0.1
+Pygments==2.18.0
+Send2Trash==1.8.3
+Werkzeug==3.0.3
+absl-py==2.1.0
+accelerate==0.30.0
+aiohttp==3.9.5
+aiosignal==1.3.1
+aniso8601==9.0.1
+annotated-types==0.6.0
+anyio==4.3.0
+argbind==0.3.7
+argon2-cffi-bindings==21.2.0
+argon2-cffi==23.1.0
+arrow==1.3.0
+asttokens==2.4.1
+async-lru==2.0.4
+attrs==23.2.0
+audioread==3.0.1
+beautifulsoup4==4.12.3
+bidict==0.23.1
+bitsandbytes==0.43.1
+bleach==6.1.0
+certifi==2024.2.2
+cffi==1.16.0
+charset-normalizer==3.3.2
+click==8.1.7
+coloredlogs==14.0
+comm==0.2.2
+contourpy==1.2.1
+cycler==0.12.1
+datasets==2.19.1
+debugpy==1.8.1
+decorator==5.1.1
+defusedxml==0.7.1
+descript-audio-codec==1.0.0
+descript-audiotools==0.7.2
+dill==0.3.8
+dnspython==2.3.0
+docker-pycreds==0.4.0
+docstring_parser==0.16
+editdistance==0.8.1
+einops==0.8.0
+et-xmlfile==1.1.0
+evaluate==0.4.2
+eventlet==0.36.1
+executing==2.0.1
+fastjsonschema==2.19.1
+ffmpy==0.3.2
+filelock==3.14.0
+fire==0.6.0
+flask-talisman==1.1.0
+flatten-dict==0.4.2
+fonttools==4.51.0
+fqdn==1.5.1
+frozenlist==1.4.1
+fsspec==2024.3.1
+future==1.0.0
+g2p==2.0.0
+gitdb==4.0.11
+greenlet==3.0.3
+grpcio==1.63.0
+h11==0.14.0
+httpcore==1.0.5
+httpx==0.27.0
+huggingface-hub==0.23.0
+humanfriendly==10.0
+idna==3.7
+importlib_resources==6.4.0
+ipdb==0.13.13
+ipykernel==6.29.4
+ipython==8.24.0
+isoduration==20.11.0
+itsdangerous==2.2.0
+jedi==0.19.1
+jiwer==3.0.4
+joblib==1.4.2
+json5==0.9.25
+jsonpointer==2.4
+jsonschema-specifications==2023.12.1
+jsonschema==4.22.0
+julius==0.2.7
+jupyter-events==0.10.0
+jupyter-lsp==2.2.5
+jupyter_client==8.6.1
+jupyter_core==5.7.2
+jupyter_server==2.14.0
+jupyter_server_terminals==0.5.3
+jupyterlab==4.2.0
+jupyterlab_pygments==0.3.0
+jupyterlab_server==2.27.1
+kiwisolver==1.4.5
+lazy_loader==0.4
+librosa==0.10.2
+llvmlite==0.42.0
+markdown-it-py==3.0.0
+markdown2==2.4.13
+matplotlib-inline==0.1.7
+matplotlib==3.8.4
+mdurl==0.1.2
+mistune==3.0.2
+mpmath==1.3.0
+msgpack==1.0.8
+multidict==6.0.5
+multiprocess==0.70.16
+munkres==1.1.4
+nbclient==0.10.0
+nbconvert==7.16.4
+nbformat==5.10.4
+nest-asyncio==1.6.0
+networkx==3.3
+notebook_shim==0.2.4
+numba==0.59.1
+numpy==1.26.4
+nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cudnn-cu12==8.9.2.26
+nvidia-cufft-cu12==11.0.2.54
+nvidia-curand-cu12==10.3.2.106
+nvidia-cusolver-cu12==11.4.5.107
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-nccl-cu12==2.20.5
+nvidia-nvjitlink-cu12==12.4.127
+nvidia-nvtx-cu12==12.1.105
+openpyxl==3.1.2
+overrides==7.7.0
+packaging==24.0
+pandas==2.2.2
+pandocfilters==1.5.1
+panphon==0.20.0
+parler_tts==0.1
+parso==0.8.4
+pexpect==4.9.0
+pillow==10.3.0
+pip==24.0
+platformdirs==4.2.1
+pooch==1.8.1
+prometheus_client==0.20.0
+prompt-toolkit==3.0.43
+protobuf==3.19.6
+psutil==5.9.8
+ptyprocess==0.7.0
+pure-eval==0.2.2
+pyarrow-hotfix==0.6
+pyarrow==16.0.0
+pycparser==2.22
+pydantic==2.7.1
+pydantic_core==2.18.2
+pyloudnorm==0.1.1
+pyparsing==3.1.2
+pystoi==0.4.1
+python-dateutil==2.9.0.post0
+python-engineio==4.9.0
+python-json-logger==2.0.7
+python-socketio==5.11.2
+pytz==2024.1
+pyzmq==26.0.3
+randomname==0.2.1
+rapidfuzz==3.9.0
+referencing==0.35.1
+regex==2024.4.28
+requests==2.31.0
+rfc3339-validator==0.1.4
+rfc3986-validator==0.1.1
+rich==13.7.1
+rpds-py==0.18.1
+safetensors==0.4.3
+scikit-learn==1.4.2
+scipy==1.13.0
+sentencepiece==0.2.0
+sentry-sdk==2.1.1
+setproctitle==1.3.3
+setuptools==69.5.1
+simple-websocket==1.0.0
+six==1.16.0
+smmap==5.0.1
+sniffio==1.3.1
+soundfile==0.12.1
+soupsieve==2.5
+soxr==0.3.7
+stack-data==0.6.3
+sympy==1.12
+tensorboard-data-server==0.7.2
+tensorboard==2.16.2
+termcolor==2.4.0
+terminado==0.18.1
+text-unidecode==1.3
+threadpoolctl==3.5.0
+tinycss2==1.3.0
+tokenizers==0.19.1
+torch-stoi==0.2.1
+torch==2.3.0
+torchaudio==2.3.0
+tornado==6.4
+tqdm==4.66.4
+traitlets==5.14.3
+transformers==4.41.0.dev0
+transformers==4.41.0.dev0
+triton==2.3.0
+types-python-dateutil==2.9.0.20240316
+typing_extensions==4.11.0
+tzdata==2024.1
+unicodecsv==0.14.1
+uri-template==1.3.0
+urllib3==2.2.1
+wandb==0.17.0
+wcwidth==0.2.13
+webcolors==1.13
+webencodings==0.5.1
+websocket-client==1.8.0
+wheel==0.43.0
+wsproto==1.2.0
+xxhash==3.4.1
+yarl==1.9.4

wandb/run-20240513_204644-y7fy6vtp/files/wandb-metadata.json

@@ -0,0 +1,804 @@
+{
+    "os": "Linux-5.4.0-166-generic-x86_64-with-glibc2.31",
+    "python": "3.11.9",
+    "heartbeatAt": "2024-05-13T18:46:44.664254",
+    "startedAt": "2024-05-13T18:46:44.142721",
+    "docker": null,
+    "cuda": null,
+    "args": [
+        "--model_name_or_path",
+        "parler-tts/parler_tts_mini_v0.1",
+        "--feature_extractor_name",
+        "parler-tts/dac_44khZ_8kbps",
+        "--description_tokenizer_name",
+        "parler-tts/parler_tts_mini_v0.1",
+        "--prompt_tokenizer_name",
+        "parler-tts/parler_tts_mini_v0.1",
+        "--report_to",
+        "wandb",
+        "--overwrite_output_dir",
+        "true",
+        "--train_dataset_name",
+        "sanchit-gandhi/expresso-concatenated-half-normal",
+        "--train_metadata_dataset_name",
+        "sanchit-gandhi/expresso-concatenated-half-normal-tags-mistral",
+        "--train_dataset_config_name",
+        "read",
+        "--train_split_name",
+        "train",
+        "--eval_dataset_name",
+        "sanchit-gandhi/expresso-concatenated-half-normal",
+        "--eval_metadata_dataset_name",
+        "sanchit-gandhi/expresso-concatenated-half-normal-tags-mistral",
+        "--eval_dataset_config_name",
+        "read",
+        "--eval_split_name",
+        "train",
+        "--max_eval_samples",
+        "8",
+        "--per_device_eval_batch_size",
+        "16",
+        "--target_audio_column_name",
+        "audio",
+        "--description_column_name",
+        "text_description",
+        "--prompt_column_name",
+        "text",
+        "--max_duration_in_seconds",
+        "30.0",
+        "--min_duration_in_seconds",
+        "2.0",
+        "--max_text_length",
+        "400",
+        "--preprocessing_num_workers",
+        "2",
+        "--do_train",
+        "true",
+        "--num_train_epochs",
+        "8",
+        "--gradient_accumulation_steps",
+        "8",
+        "--gradient_checkpointing",
+        "true",
+        "--per_device_train_batch_size",
+        "16",
+        "--learning_rate",
+        "0.00008",
+        "--adam_beta1",
+        "0.9",
+        "--adam_beta2",
+        "0.99",
+        "--weight_decay",
+        "0.01",
+        "--lr_scheduler_type",
+        "cosine",
+        "--warmup_steps",
+        "250",
+        "--logging_steps",
+        "2",
+        "--freeze_text_encoder",
+        "true",
+        "--audio_encoder_per_device_batch_size",
+        "4",
+        "--dtype",
+        "bfloat16",
+        "--seed",
+        "456",
+        "--output_dir",
+        "../output_dir_training_concat/",
+        "--temporary_save_to_disk",
+        "../audio_code_tmp_concat/",
+        "--save_to_disk",
+        "../tmp_dataset_audio_concat/",
+        "--dataloader_num_workers",
+        "4",
+        "--do_eval",
+        "--predict_with_generate",
+        "--include_inputs_for_metrics",
+        "--save_strategy",
+        "steps",
+        "--save_steps",
+        "72",
+        "--evaluation_strategy",
+        "epoch",
+        "--save_total_limit",
+        "5",
+        "--group_by_length",
+        "true"
+    ],
+    "state": "running",
+    "program": "/raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/./run_parler_tts_training.py",
+    "codePathLocal": "run_parler_tts_training.py",
+    "codePath": "run_parler_tts_training.py",
+    "git": {
+        "remote": "https://huggingface.co/sanchit-gandhi/parler-tts-mini-v0.1-expresso-concatenated",
+        "commit": "99f75adb5e13ee0ad87cce8deb3e71adc370b10e"
+    },
+    "email": "sanchit@huggingface.co",
+    "root": "/raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated",
+    "host": "hf-dgx-01",
+    "username": "sanchit",
+    "executable": "/home/sanchit/miniconda3/envs/venv/bin/python",
+    "cpu_count": 64,
+    "cpu_count_logical": 128,
+    "cpu_freq": {
+        "current": 2713.9304453124996,
+        "min": 1500.0,
+        "max": 2250.0
+    },
+    "cpu_freq_per_core": [
+        {
+            "current": 2076.148,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1783.251,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1793.241,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1796.012,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2273.163,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1749.542,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1747.968,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1749.532,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3151.024,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2454.223,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1717.418,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2440.313,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1726.936,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3285.322,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1665.33,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1714.033,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1666.499,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3313.14,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3148.73,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1667.633,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3314.333,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1665.824,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1663.97,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1669.416,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1711.485,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2435.786,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1712.451,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1713.291,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2260.754,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1713.396,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1716.495,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1716.727,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1715.159,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1666.66,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2327.712,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1716.046,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1715.048,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3233.973,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1713.251,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3198.61,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1713.522,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1716.503,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1716.921,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2449.66,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1668.915,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2568.264,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1668.642,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3314.602,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1713.343,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1710.688,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2995.061,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1714.724,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3259.362,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1711.982,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1723.149,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1704.811,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2298.41,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1732.061,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1697.486,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2387.266,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1791.846,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1794.417,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1794.03,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1792.18,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2259.304,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2465.862,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2463.001,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2459.093,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2997.527,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2331.651,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2591.488,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2443.283,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2415.125,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2461.32,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1885.452,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1808.205,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2520.778,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3303.109,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2475.55,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2536.741,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3148.399,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3332.648,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2543.883,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1667.131,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3332.767,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1666.858,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1666.153,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1665.151,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1713.213,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2507.482,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1713.066,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1670.247,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2279.452,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1712.054,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1713.383,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1711.275,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1715.372,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1714.004,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3181.039,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1714.611,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1714.271,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2529.667,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1714.294,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2456.025,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1712.501,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1717.28,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1712.863,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2453.728,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1666.803,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1668.423,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1666.817,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3335.646,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1664.711,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1750.001,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3334.823,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1659.52,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3342.577,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1661.087,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2210.973,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1661.877,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1929.122,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2363.412,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2012.647,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3318.307,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1792.637,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1794.88,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1795.26,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1796.177,
+            "min": 1500.0,
+            "max": 2250.0
+        }
+    ],
+    "disk": {
+        "/": {
+            "total": 1757.8785285949707,
+            "used": 1663.5005645751953
+        }
+    },
+    "gpu": "NVIDIA A100-SXM4-80GB",
+    "gpu_count": 5,
+    "gpu_devices": [
+        {
+            "name": "NVIDIA A100-SXM4-80GB",
+            "memory_total": 85899345920
+        },
+        {
+            "name": "NVIDIA A100-SXM4-80GB",
+            "memory_total": 85899345920
+        },
+        {
+            "name": "NVIDIA A100-SXM4-80GB",
+            "memory_total": 85899345920
+        },
+        {
+            "name": "NVIDIA DGX Display",
+            "memory_total": 4294967296
+        },
+        {
+            "name": "NVIDIA A100-SXM4-80GB",
+            "memory_total": 85899345920
+        }
+    ],
+    "memory": {
+        "total": 503.5396919250488
+    }
+}

wandb/run-20240513_204644-y7fy6vtp/files/wandb-summary.json

@@ -0,0 +1 @@
+{"_wandb": {"runtime": 228}}

wandb/run-20240513_204644-y7fy6vtp/logs/debug-internal.log

@@ -0,0 +1,465 @@
+2024-05-13 20:46:44,150 INFO    StreamThr :1244488 [internal.py:wandb_internal():85] W&B internal server running at pid: 1244488, started at: 2024-05-13 20:46:44.149628
+2024-05-13 20:46:44,152 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status
+2024-05-13 20:46:44,154 INFO    WriterThread:1244488 [datastore.py:open_for_write():87] open: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/run-y7fy6vtp.wandb
+2024-05-13 20:46:44,156 DEBUG   SenderThread:1244488 [sender.py:send():378] send: header
+2024-05-13 20:46:44,156 DEBUG   SenderThread:1244488 [sender.py:send():378] send: run
+2024-05-13 20:46:44,526 INFO    SenderThread:1244488 [dir_watcher.py:__init__():211] watching files in: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files
+2024-05-13 20:46:44,526 INFO    SenderThread:1244488 [sender.py:_start_run_threads():1123] run started: y7fy6vtp with start time 1715626004.147997
+2024-05-13 20:46:44,532 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: check_version
+2024-05-13 20:46:44,532 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: check_version
+2024-05-13 20:46:44,596 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: run_start
+2024-05-13 20:46:44,626 DEBUG   HandlerThread:1244488 [system_info.py:__init__():26] System info init
+2024-05-13 20:46:44,626 DEBUG   HandlerThread:1244488 [system_info.py:__init__():41] System info init done
+2024-05-13 20:46:44,626 INFO    HandlerThread:1244488 [system_monitor.py:start():194] Starting system monitor
+2024-05-13 20:46:44,626 INFO    SystemMonitor:1244488 [system_monitor.py:_start():158] Starting system asset monitoring threads
+2024-05-13 20:46:44,627 INFO    HandlerThread:1244488 [system_monitor.py:probe():214] Collecting system info
+2024-05-13 20:46:44,627 INFO    SystemMonitor:1244488 [interfaces.py:start():188] Started cpu monitoring
+2024-05-13 20:46:44,628 INFO    SystemMonitor:1244488 [interfaces.py:start():188] Started disk monitoring
+2024-05-13 20:46:44,628 INFO    SystemMonitor:1244488 [interfaces.py:start():188] Started gpu monitoring
+2024-05-13 20:46:44,629 INFO    SystemMonitor:1244488 [interfaces.py:start():188] Started memory monitoring
+2024-05-13 20:46:44,630 INFO    SystemMonitor:1244488 [interfaces.py:start():188] Started network monitoring
+2024-05-13 20:46:44,664 DEBUG   HandlerThread:1244488 [system_info.py:probe():150] Probing system
+2024-05-13 20:46:44,665 DEBUG   HandlerThread:1244488 [system_info.py:_probe_git():135] Probing git
+2024-05-13 20:46:44,670 DEBUG   HandlerThread:1244488 [system_info.py:_probe_git():143] Probing git done
+2024-05-13 20:46:44,670 DEBUG   HandlerThread:1244488 [system_info.py:probe():198] Probing system done
+2024-05-13 20:46:44,670 DEBUG   HandlerThread:1244488 [system_monitor.py:probe():223] {'os': 'Linux-5.4.0-166-generic-x86_64-with-glibc2.31', 'python': '3.11.9', 'heartbeatAt': '2024-05-13T18:46:44.664254', 'startedAt': '2024-05-13T18:46:44.142721', 'docker': None, 'cuda': None, 'args': ('--model_name_or_path', 'parler-tts/parler_tts_mini_v0.1', '--feature_extractor_name', 'parler-tts/dac_44khZ_8kbps', '--description_tokenizer_name', 'parler-tts/parler_tts_mini_v0.1', '--prompt_tokenizer_name', 'parler-tts/parler_tts_mini_v0.1', '--report_to', 'wandb', '--overwrite_output_dir', 'true', '--train_dataset_name', 'sanchit-gandhi/expresso-concatenated-half-normal', '--train_metadata_dataset_name', 'sanchit-gandhi/expresso-concatenated-half-normal-tags-mistral', '--train_dataset_config_name', 'read', '--train_split_name', 'train', '--eval_dataset_name', 'sanchit-gandhi/expresso-concatenated-half-normal', '--eval_metadata_dataset_name', 'sanchit-gandhi/expresso-concatenated-half-normal-tags-mistral', '--eval_dataset_config_name', 'read', '--eval_split_name', 'train', '--max_eval_samples', '8', '--per_device_eval_batch_size', '16', '--target_audio_column_name', 'audio', '--description_column_name', 'text_description', '--prompt_column_name', 'text', '--max_duration_in_seconds', '30.0', '--min_duration_in_seconds', '2.0', '--max_text_length', '400', '--preprocessing_num_workers', '2', '--do_train', 'true', '--num_train_epochs', '8', '--gradient_accumulation_steps', '8', '--gradient_checkpointing', 'true', '--per_device_train_batch_size', '16', '--learning_rate', '0.00008', '--adam_beta1', '0.9', '--adam_beta2', '0.99', '--weight_decay', '0.01', '--lr_scheduler_type', 'cosine', '--warmup_steps', '250', '--logging_steps', '2', '--freeze_text_encoder', 'true', '--audio_encoder_per_device_batch_size', '4', '--dtype', 'bfloat16', '--seed', '456', '--output_dir', '../output_dir_training_concat/', '--temporary_save_to_disk', '../audio_code_tmp_concat/', '--save_to_disk', '../tmp_dataset_audio_concat/', '--dataloader_num_workers', '4', '--do_eval', '--predict_with_generate', '--include_inputs_for_metrics', '--save_strategy', 'steps', '--save_steps', '72', '--evaluation_strategy', 'epoch', '--save_total_limit', '5', '--group_by_length', 'true'), 'state': 'running', 'program': '/raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/./run_parler_tts_training.py', 'codePathLocal': 'run_parler_tts_training.py', 'codePath': 'run_parler_tts_training.py', 'git': {'remote': 'https://huggingface.co/sanchit-gandhi/parler-tts-mini-v0.1-expresso-concatenated', 'commit': '99f75adb5e13ee0ad87cce8deb3e71adc370b10e'}, 'email': 'sanchit@huggingface.co', 'root': '/raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated', 'host': 'hf-dgx-01', 'username': 'sanchit', 'executable': '/home/sanchit/miniconda3/envs/venv/bin/python', 'cpu_count': 64, 'cpu_count_logical': 128, 'cpu_freq': {'current': 2713.9304453124996, 'min': 1500.0, 'max': 2250.0}, 'cpu_freq_per_core': [{'current': 2076.148, 'min': 1500.0, 'max': 2250.0}, {'current': 1783.251, 'min': 1500.0, 'max': 2250.0}, {'current': 1793.241, 'min': 1500.0, 'max': 2250.0}, {'current': 1796.012, 'min': 1500.0, 'max': 2250.0}, {'current': 2273.163, 'min': 1500.0, 'max': 2250.0}, {'current': 1749.542, 'min': 1500.0, 'max': 2250.0}, {'current': 1747.968, 'min': 1500.0, 'max': 2250.0}, {'current': 1749.532, 'min': 1500.0, 'max': 2250.0}, {'current': 3151.024, 'min': 1500.0, 'max': 2250.0}, {'current': 2454.223, 'min': 1500.0, 'max': 2250.0}, {'current': 1717.418, 'min': 1500.0, 'max': 2250.0}, {'current': 2440.313, 'min': 1500.0, 'max': 2250.0}, {'current': 1726.936, 'min': 1500.0, 'max': 2250.0}, {'current': 3285.322, 'min': 1500.0, 'max': 2250.0}, {'current': 1665.33, 'min': 1500.0, 'max': 2250.0}, {'current': 1714.033, 'min': 1500.0, 'max': 2250.0}, {'current': 1666.499, 'min': 1500.0, 'max': 2250.0}, {'current': 3313.14, 'min': 1500.0, 'max': 2250.0}, {'current': 3148.73, 'min': 1500.0, 'max': 2250.0}, {'current': 1667.633, 'min': 1500.0, 'max': 2250.0}, {'current': 3314.333, 'min': 1500.0, 'max': 2250.0}, {'current': 1665.824, 'min': 1500.0, 'max': 2250.0}, {'current': 1663.97, 'min': 1500.0, 'max': 2250.0}, {'current': 1669.416, 'min': 1500.0, 'max': 2250.0}, {'current': 1711.485, 'min': 1500.0, 'max': 2250.0}, {'current': 2435.786, 'min': 1500.0, 'max': 2250.0}, {'current': 1712.451, 'min': 1500.0, 'max': 2250.0}, {'current': 1713.291, 'min': 1500.0, 'max': 2250.0}, {'current': 2260.754, 'min': 1500.0, 'max': 2250.0}, {'current': 1713.396, 'min': 1500.0, 'max': 2250.0}, {'current': 1716.495, 'min': 1500.0, 'max': 2250.0}, {'current': 1716.727, 'min': 1500.0, 'max': 2250.0}, {'current': 1715.159, 'min': 1500.0, 'max': 2250.0}, {'current': 1666.66, 'min': 1500.0, 'max': 2250.0}, {'current': 2327.712, 'min': 1500.0, 'max': 2250.0}, {'current': 1716.046, 'min': 1500.0, 'max': 2250.0}, {'current': 1715.048, 'min': 1500.0, 'max': 2250.0}, {'current': 3233.973, 'min': 1500.0, 'max': 2250.0}, {'current': 1713.251, 'min': 1500.0, 'max': 2250.0}, {'current': 3198.61, 'min': 1500.0, 'max': 2250.0}, {'current': 1713.522, 'min': 1500.0, 'max': 2250.0}, {'current': 1716.503, 'min': 1500.0, 'max': 2250.0}, {'current': 1716.921, 'min': 1500.0, 'max': 2250.0}, {'current': 2449.66, 'min': 1500.0, 'max': 2250.0}, {'current': 1668.915, 'min': 1500.0, 'max': 2250.0}, {'current': 2568.264, 'min': 1500.0, 'max': 2250.0}, {'current': 1668.642, 'min': 1500.0, 'max': 2250.0}, {'current': 3314.602, 'min': 1500.0, 'max': 2250.0}, {'current': 1713.343, 'min': 1500.0, 'max': 2250.0}, {'current': 1710.688, 'min': 1500.0, 'max': 2250.0}, {'current': 2995.061, 'min': 1500.0, 'max': 2250.0}, {'current': 1714.724, 'min': 1500.0, 'max': 2250.0}, {'current': 3259.362, 'min': 1500.0, 'max': 2250.0}, {'current': 1711.982, 'min': 1500.0, 'max': 2250.0}, {'current': 1723.149, 'min': 1500.0, 'max': 2250.0}, {'current': 1704.811, 'min': 1500.0, 'max': 2250.0}, {'current': 2298.41, 'min': 1500.0, 'max': 2250.0}, {'current': 1732.061, 'min': 1500.0, 'max': 2250.0}, {'current': 1697.486, 'min': 1500.0, 'max': 2250.0}, {'current': 2387.266, 'min': 1500.0, 'max': 2250.0}, {'current': 1791.846, 'min': 1500.0, 'max': 2250.0}, {'current': 1794.417, 'min': 1500.0, 'max': 2250.0}, {'current': 1794.03, 'min': 1500.0, 'max': 2250.0}, {'current': 1792.18, 'min': 1500.0, 'max': 2250.0}, {'current': 2259.304, 'min': 1500.0, 'max': 2250.0}, {'current': 2465.862, 'min': 1500.0, 'max': 2250.0}, {'current': 2463.001, 'min': 1500.0, 'max': 2250.0}, {'current': 2459.093, 'min': 1500.0, 'max': 2250.0}, {'current': 2997.527, 'min': 1500.0, 'max': 2250.0}, {'current': 2331.651, 'min': 1500.0, 'max': 2250.0}, {'current': 2591.488, 'min': 1500.0, 'max': 2250.0}, {'current': 2443.283, 'min': 1500.0, 'max': 2250.0}, {'current': 2415.125, 'min': 1500.0, 'max': 2250.0}, {'current': 2461.32, 'min': 1500.0, 'max': 2250.0}, {'current': 1885.452, 'min': 1500.0, 'max': 2250.0}, {'current': 1808.205, 'min': 1500.0, 'max': 2250.0}, {'current': 2520.778, 'min': 1500.0, 'max': 2250.0}, {'current': 3303.109, 'min': 1500.0, 'max': 2250.0}, {'current': 2475.55, 'min': 1500.0, 'max': 2250.0}, {'current': 2536.741, 'min': 1500.0, 'max': 2250.0}, {'current': 3148.399, 'min': 1500.0, 'max': 2250.0}, {'current': 3332.648, 'min': 1500.0, 'max': 2250.0}, {'current': 2543.883, 'min': 1500.0, 'max': 2250.0}, {'current': 1667.131, 'min': 1500.0, 'max': 2250.0}, {'current': 3332.767, 'min': 1500.0, 'max': 2250.0}, {'current': 1666.858, 'min': 1500.0, 'max': 2250.0}, {'current': 1666.153, 'min': 1500.0, 'max': 2250.0}, {'current': 1665.151, 'min': 1500.0, 'max': 2250.0}, {'current': 1713.213, 'min': 1500.0, 'max': 2250.0}, {'current': 2507.482, 'min': 1500.0, 'max': 2250.0}, {'current': 1713.066, 'min': 1500.0, 'max': 2250.0}, {'current': 1670.247, 'min': 1500.0, 'max': 2250.0}, {'current': 2279.452, 'min': 1500.0, 'max': 2250.0}, {'current': 1712.054, 'min': 1500.0, 'max': 2250.0}, {'current': 1713.383, 'min': 1500.0, 'max': 2250.0}, {'current': 1711.275, 'min': 1500.0, 'max': 2250.0}, {'current': 1715.372, 'min': 1500.0, 'max': 2250.0}, {'current': 1714.004, 'min': 1500.0, 'max': 2250.0}, {'current': 3181.039, 'min': 1500.0, 'max': 2250.0}, {'current': 1714.611, 'min': 1500.0, 'max': 2250.0}, {'current': 1714.271, 'min': 1500.0, 'max': 2250.0}, {'current': 2529.667, 'min': 1500.0, 'max': 2250.0}, {'current': 1714.294, 'min': 1500.0, 'max': 2250.0}, {'current': 2456.025, 'min': 1500.0, 'max': 2250.0}, {'current': 1712.501, 'min': 1500.0, 'max': 2250.0}, {'current': 1717.28, 'min': 1500.0, 'max': 2250.0}, {'current': 1712.863, 'min': 1500.0, 'max': 2250.0}, {'current': 2453.728, 'min': 1500.0, 'max': 2250.0}, {'current': 1666.803, 'min': 1500.0, 'max': 2250.0}, {'current': 1668.423, 'min': 1500.0, 'max': 2250.0}, {'current': 1666.817, 'min': 1500.0, 'max': 2250.0}, {'current': 3335.646, 'min': 1500.0, 'max': 2250.0}, {'current': 1664.711, 'min': 1500.0, 'max': 2250.0}, {'current': 1750.001, 'min': 1500.0, 'max': 2250.0}, {'current': 3334.823, 'min': 1500.0, 'max': 2250.0}, {'current': 1659.52, 'min': 1500.0, 'max': 2250.0}, {'current': 3342.577, 'min': 1500.0, 'max': 2250.0}, {'current': 1661.087, 'min': 1500.0, 'max': 2250.0}, {'current': 2210.973, 'min': 1500.0, 'max': 2250.0}, {'current': 1661.877, 'min': 1500.0, 'max': 2250.0}, {'current': 1929.122, 'min': 1500.0, 'max': 2250.0}, {'current': 2363.412, 'min': 1500.0, 'max': 2250.0}, {'current': 2012.647, 'min': 1500.0, 'max': 2250.0}, {'current': 3318.307, 'min': 1500.0, 'max': 2250.0}, {'current': 1792.637, 'min': 1500.0, 'max': 2250.0}, {'current': 1794.88, 'min': 1500.0, 'max': 2250.0}, {'current': 1795.26, 'min': 1500.0, 'max': 2250.0}, {'current': 1796.177, 'min': 1500.0, 'max': 2250.0}], 'disk': {'/': {'total': 1757.8785285949707, 'used': 1663.5005645751953}}, 'gpu': 'NVIDIA A100-SXM4-80GB', 'gpu_count': 5, 'gpu_devices': [{'name': 'NVIDIA A100-SXM4-80GB', 'memory_total': 85899345920}, {'name': 'NVIDIA A100-SXM4-80GB', 'memory_total': 85899345920}, {'name': 'NVIDIA A100-SXM4-80GB', 'memory_total': 85899345920}, {'name': 'NVIDIA DGX Display', 'memory_total': 4294967296}, {'name': 'NVIDIA A100-SXM4-80GB', 'memory_total': 85899345920}], 'memory': {'total': 503.5396919250488}}
+2024-05-13 20:46:44,671 INFO    HandlerThread:1244488 [system_monitor.py:probe():224] Finished collecting system info
+2024-05-13 20:46:44,671 INFO    HandlerThread:1244488 [system_monitor.py:probe():227] Publishing system info
+2024-05-13 20:46:44,671 DEBUG   HandlerThread:1244488 [system_info.py:_save_conda():207] Saving list of conda packages installed into the current environment
+2024-05-13 20:46:44,685 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:46:44,700 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:46:45,528 INFO    Thread-12 :1244488 [dir_watcher.py:_on_file_created():271] file/dir created: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/conda-environment.yaml
+2024-05-13 20:46:46,722 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:46:46,733 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:46:47,041 DEBUG   HandlerThread:1244488 [system_info.py:_save_conda():222] Saving conda packages done
+2024-05-13 20:46:47,042 INFO    HandlerThread:1244488 [system_monitor.py:probe():229] Finished publishing system info
+2024-05-13 20:46:47,050 DEBUG   SenderThread:1244488 [sender.py:send():378] send: files
+2024-05-13 20:46:47,051 INFO    SenderThread:1244488 [sender.py:_save_file():1389] saving file wandb-metadata.json with policy now
+2024-05-13 20:46:47,164 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: python_packages
+2024-05-13 20:46:47,164 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:46:47,165 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: python_packages
+2024-05-13 20:46:47,167 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:46:47,399 DEBUG   SenderThread:1244488 [sender.py:send():378] send: telemetry
+2024-05-13 20:46:47,399 DEBUG   SenderThread:1244488 [sender.py:send():378] send: config
+2024-05-13 20:46:47,527 INFO    Thread-12 :1244488 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/conda-environment.yaml
+2024-05-13 20:46:47,527 INFO    Thread-12 :1244488 [dir_watcher.py:_on_file_created():271] file/dir created: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/wandb-metadata.json
+2024-05-13 20:46:47,527 INFO    Thread-12 :1244488 [dir_watcher.py:_on_file_created():271] file/dir created: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/requirements.txt
+2024-05-13 20:46:47,527 INFO    Thread-12 :1244488 [dir_watcher.py:_on_file_created():271] file/dir created: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/output.log
+2024-05-13 20:46:47,577 INFO    wandb-upload_0:1244488 [upload_job.py:push():130] Uploaded file /tmp/tmpub0aqma1wandb/xhoqu8pi-wandb-metadata.json
+2024-05-13 20:46:49,371 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:46:49,527 INFO    Thread-12 :1244488 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/output.log
+2024-05-13 20:46:49,761 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:46:49,774 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:46:51,528 INFO    Thread-12 :1244488 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/output.log
+2024-05-13 20:46:51,799 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:46:51,809 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:46:54,372 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:46:54,928 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:46:54,950 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:46:56,968 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:46:56,979 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:46:59,372 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:46:59,995 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:00,017 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:02,037 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:02,060 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:02,164 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:47:02,165 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:47:05,234 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:05,248 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:05,342 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:47:07,294 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:07,305 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:10,342 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:47:10,471 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:10,483 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:12,502 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:12,511 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:14,539 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:15,347 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:47:15,535 INFO    Thread-12 :1244488 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/config.yaml
+2024-05-13 20:47:15,597 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:17,163 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:47:17,164 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:47:17,617 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:17,627 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:20,776 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:20,799 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:21,373 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:47:24,480 ERROR   gpu       :1244488 [interfaces.py:aggregate():159] Failed to serialize metric: division by zero
+2024-05-13 20:47:24,503 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:24,513 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:26,374 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:47:26,539 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:26,549 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:29,415 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:29,435 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:31,375 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:47:31,471 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:31,494 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:32,163 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:47:32,164 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:47:34,504 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:34,522 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:36,590 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:36,609 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:37,373 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:47:38,669 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:38,682 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:41,557 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:41,584 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:42,374 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:47:43,631 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:43,652 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:44,630 DEBUG   SystemMonitor:1244488 [system_monitor.py:_start():172] Starting system metrics aggregation loop
+2024-05-13 20:47:44,634 DEBUG   SenderThread:1244488 [sender.py:send():378] send: stats
+2024-05-13 20:47:46,570 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:46,604 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:47,164 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:47:47,164 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:47:48,372 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:47:48,627 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:48,647 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:51,667 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:51,686 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:53,373 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:47:53,731 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:53,745 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:56,603 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:56,614 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:58,374 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:47:58,651 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:47:58,660 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:02,165 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:48:02,165 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:48:02,441 ERROR   gpu       :1244488 [interfaces.py:aggregate():159] Failed to serialize metric: division by zero
+2024-05-13 20:48:03,327 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:03,354 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:03,406 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:48:05,386 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:05,421 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:08,407 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:48:08,479 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:08,498 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:10,520 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:10,533 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:13,408 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:48:13,506 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:13,529 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:14,637 DEBUG   SenderThread:1244488 [sender.py:send():378] send: stats
+2024-05-13 20:48:15,573 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:15,595 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:17,164 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:48:17,165 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:48:18,553 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:18,576 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:19,398 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:48:20,630 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:20,645 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:23,668 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:23,707 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:24,399 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:48:25,750 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:25,769 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:28,690 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:28,701 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:29,400 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:48:30,752 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:30,766 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:32,164 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:48:32,165 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:48:33,779 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:33,801 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:35,390 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:48:35,833 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:35,857 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:38,883 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:38,904 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:40,391 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:48:42,674 ERROR   gpu       :1244488 [interfaces.py:aggregate():159] Failed to serialize metric: division by zero
+2024-05-13 20:48:42,717 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:42,728 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:44,638 DEBUG   SenderThread:1244488 [sender.py:send():378] send: stats
+2024-05-13 20:48:45,640 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:48:45,792 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:45,830 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:47,165 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:48:47,166 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:48:47,862 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:47,885 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:50,784 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:50,806 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:51,392 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:48:52,825 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:52,839 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:55,925 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:55,958 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:56,393 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:48:57,986 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:48:58,004 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:01,057 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:01,082 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:01,394 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:49:02,165 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:49:02,165 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:49:03,103 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:03,126 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:06,046 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:06,069 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:07,371 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:49:08,089 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:08,099 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:11,174 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:11,186 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:12,372 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:49:13,203 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:13,213 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:14,641 DEBUG   SenderThread:1244488 [sender.py:send():378] send: stats
+2024-05-13 20:49:16,181 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:16,195 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:17,165 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:49:17,165 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:49:18,217 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:18,228 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:18,360 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:49:21,690 ERROR   gpu       :1244488 [interfaces.py:aggregate():159] Failed to serialize metric: division by zero
+2024-05-13 20:49:21,712 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:21,934 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:23,361 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:49:24,088 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:24,097 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:26,124 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:26,134 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:28,362 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:49:28,618 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:28,634 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:30,672 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:30,685 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:32,165 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:49:32,165 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:49:33,142 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:33,160 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:33,398 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:49:35,183 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:35,200 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:37,936 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:37,992 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:38,398 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:49:40,036 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:40,045 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:42,607 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:42,624 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:43,399 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:49:44,643 DEBUG   SenderThread:1244488 [sender.py:send():378] send: stats
+2024-05-13 20:49:44,677 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:44,693 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:47,165 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:49:47,165 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:49:47,255 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:47,292 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:49,311 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:49,328 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:49:49,336 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:51,888 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:51,906 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:54,329 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:49:54,371 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:54,403 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:58,505 ERROR   gpu       :1244488 [interfaces.py:aggregate():159] Failed to serialize metric: division by zero
+2024-05-13 20:49:59,014 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:59,023 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:49:59,330 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:50:01,042 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:01,076 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:02,165 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:50:02,165 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:50:03,671 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:03,687 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:04,400 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:50:05,735 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:05,745 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:08,359 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:08,383 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:09,405 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:50:10,443 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:10,463 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:13,159 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:13,196 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:14,405 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:50:14,646 DEBUG   SenderThread:1244488 [sender.py:send():378] send: stats
+2024-05-13 20:50:15,219 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:15,230 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:17,165 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:50:17,165 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:50:17,757 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:17,769 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:19,788 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:19,798 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:20,116 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:50:21,592 INFO    Thread-12 :1244488 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/output.log
+2024-05-13 20:50:22,219 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:22,235 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:24,253 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:24,262 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:25,150 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:50:25,593 INFO    Thread-12 :1244488 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/output.log
+2024-05-13 20:50:26,744 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:26,754 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:27,593 INFO    Thread-12 :1244488 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/output.log
+2024-05-13 20:50:28,771 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:28,784 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:29,594 INFO    Thread-12 :1244488 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/output.log
+2024-05-13 20:50:30,834 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:50:31,305 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:31,314 ERROR   gpu       :1244488 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:50:31,594 INFO    Thread-12 :1244488 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/output.log
+2024-05-13 20:50:32,165 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:50:32,165 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:50:33,134 DEBUG   SenderThread:1244488 [sender.py:send():378] send: exit
+2024-05-13 20:50:33,134 INFO    SenderThread:1244488 [sender.py:send_exit():585] handling exit code: 1
+2024-05-13 20:50:33,134 INFO    SenderThread:1244488 [sender.py:send_exit():587] handling runtime: 228
+2024-05-13 20:50:33,134 INFO    SenderThread:1244488 [sender.py:_save_file():1389] saving file wandb-summary.json with policy end
+2024-05-13 20:50:33,134 INFO    SenderThread:1244488 [sender.py:send_exit():593] send defer
+2024-05-13 20:50:33,135 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:50:33,135 INFO    HandlerThread:1244488 [handler.py:handle_request_defer():184] handle defer: 0
+2024-05-13 20:50:33,135 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:50:33,135 INFO    SenderThread:1244488 [sender.py:send_request_defer():609] handle sender defer: 0
+2024-05-13 20:50:33,135 INFO    SenderThread:1244488 [sender.py:transition_state():613] send defer: 1
+2024-05-13 20:50:33,135 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:50:33,135 INFO    HandlerThread:1244488 [handler.py:handle_request_defer():184] handle defer: 1
+2024-05-13 20:50:33,135 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:50:33,135 INFO    SenderThread:1244488 [sender.py:send_request_defer():609] handle sender defer: 1
+2024-05-13 20:50:33,135 INFO    SenderThread:1244488 [sender.py:transition_state():613] send defer: 2
+2024-05-13 20:50:33,135 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:50:33,135 INFO    HandlerThread:1244488 [handler.py:handle_request_defer():184] handle defer: 2
+2024-05-13 20:50:33,136 INFO    HandlerThread:1244488 [system_monitor.py:finish():203] Stopping system monitor
+2024-05-13 20:50:33,136 DEBUG   SystemMonitor:1244488 [system_monitor.py:_start():179] Finished system metrics aggregation loop
+2024-05-13 20:50:33,136 DEBUG   SystemMonitor:1244488 [system_monitor.py:_start():183] Publishing last batch of metrics
+2024-05-13 20:50:33,139 INFO    HandlerThread:1244488 [interfaces.py:finish():200] Joined cpu monitor
+2024-05-13 20:50:33,139 INFO    HandlerThread:1244488 [interfaces.py:finish():200] Joined disk monitor
+2024-05-13 20:50:33,595 INFO    Thread-12 :1244488 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/output.log
+2024-05-13 20:50:33,595 INFO    Thread-12 :1244488 [dir_watcher.py:_on_file_created():271] file/dir created: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/wandb-summary.json
+2024-05-13 20:50:34,732 ERROR   gpu       :1244488 [interfaces.py:aggregate():159] Failed to serialize metric: division by zero
+2024-05-13 20:50:34,732 INFO    HandlerThread:1244488 [interfaces.py:finish():200] Joined gpu monitor
+2024-05-13 20:50:34,732 INFO    HandlerThread:1244488 [interfaces.py:finish():200] Joined memory monitor
+2024-05-13 20:50:34,733 INFO    HandlerThread:1244488 [interfaces.py:finish():200] Joined network monitor
+2024-05-13 20:50:34,733 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: poll_exit
+2024-05-13 20:50:34,734 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:50:34,734 INFO    SenderThread:1244488 [sender.py:send_request_defer():609] handle sender defer: 2
+2024-05-13 20:50:34,735 INFO    SenderThread:1244488 [sender.py:transition_state():613] send defer: 3
+2024-05-13 20:50:34,735 DEBUG   SenderThread:1244488 [sender.py:send():378] send: stats
+2024-05-13 20:50:34,736 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: poll_exit
+2024-05-13 20:50:34,736 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:50:34,736 INFO    HandlerThread:1244488 [handler.py:handle_request_defer():184] handle defer: 3
+2024-05-13 20:50:34,736 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:50:34,736 INFO    SenderThread:1244488 [sender.py:send_request_defer():609] handle sender defer: 3
+2024-05-13 20:50:34,736 INFO    SenderThread:1244488 [sender.py:transition_state():613] send defer: 4
+2024-05-13 20:50:34,736 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:50:34,736 INFO    HandlerThread:1244488 [handler.py:handle_request_defer():184] handle defer: 4
+2024-05-13 20:50:34,736 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:50:34,737 INFO    SenderThread:1244488 [sender.py:send_request_defer():609] handle sender defer: 4
+2024-05-13 20:50:34,737 INFO    SenderThread:1244488 [sender.py:transition_state():613] send defer: 5
+2024-05-13 20:50:34,737 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:50:34,737 INFO    HandlerThread:1244488 [handler.py:handle_request_defer():184] handle defer: 5
+2024-05-13 20:50:34,737 DEBUG   SenderThread:1244488 [sender.py:send():378] send: summary
+2024-05-13 20:50:34,737 INFO    SenderThread:1244488 [sender.py:_save_file():1389] saving file wandb-summary.json with policy end
+2024-05-13 20:50:34,737 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:50:34,737 INFO    SenderThread:1244488 [sender.py:send_request_defer():609] handle sender defer: 5
+2024-05-13 20:50:34,737 INFO    SenderThread:1244488 [sender.py:transition_state():613] send defer: 6
+2024-05-13 20:50:34,738 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:50:34,738 INFO    HandlerThread:1244488 [handler.py:handle_request_defer():184] handle defer: 6
+2024-05-13 20:50:34,738 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:50:34,738 INFO    SenderThread:1244488 [sender.py:send_request_defer():609] handle sender defer: 6
+2024-05-13 20:50:34,738 INFO    SenderThread:1244488 [sender.py:transition_state():613] send defer: 7
+2024-05-13 20:50:34,738 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:50:34,738 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:50:34,738 INFO    HandlerThread:1244488 [handler.py:handle_request_defer():184] handle defer: 7
+2024-05-13 20:50:34,738 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:50:34,738 INFO    SenderThread:1244488 [sender.py:send_request_defer():609] handle sender defer: 7
+2024-05-13 20:50:35,134 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: poll_exit
+2024-05-13 20:50:35,540 INFO    SenderThread:1244488 [sender.py:transition_state():613] send defer: 8
+2024-05-13 20:50:35,540 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: poll_exit
+2024-05-13 20:50:35,540 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:50:35,541 INFO    HandlerThread:1244488 [handler.py:handle_request_defer():184] handle defer: 8
+2024-05-13 20:50:35,541 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:50:35,541 INFO    SenderThread:1244488 [sender.py:send_request_defer():609] handle sender defer: 8
+2024-05-13 20:50:35,541 INFO    SenderThread:1244488 [job_builder.py:build():432] Attempting to build job artifact
+2024-05-13 20:50:35,541 INFO    SenderThread:1244488 [job_builder.py:_get_source_type():565] is repo sourced job
+2024-05-13 20:50:35,565 INFO    SenderThread:1244488 [job_builder.py:build():541] adding wandb-job metadata file
+2024-05-13 20:50:35,566 INFO    SenderThread:1244488 [sender.py:transition_state():613] send defer: 9
+2024-05-13 20:50:35,567 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:50:35,567 DEBUG   SenderThread:1244488 [sender.py:send():378] send: artifact
+2024-05-13 20:50:35,567 INFO    HandlerThread:1244488 [handler.py:handle_request_defer():184] handle defer: 9
+2024-05-13 20:50:35,596 INFO    Thread-12 :1244488 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/wandb-summary.json
+2024-05-13 20:50:35,596 INFO    Thread-12 :1244488 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/output.log
+2024-05-13 20:50:36,135 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: poll_exit
+2024-05-13 20:50:36,629 INFO    wandb-upload_1:1244488 [upload_job.py:push():88] Uploaded file /tmp/tmpnjo96axd/wandb-job.json
+2024-05-13 20:50:36,674 INFO    wandb-upload_0:1244488 [upload_job.py:push():88] Uploaded file /home/sanchit/.local/share/wandb/artifacts/staging/tmp8rqc3i5s
+2024-05-13 20:50:37,629 INFO    SenderThread:1244488 [sender.py:send_artifact():1467] sent artifact job-https___huggingface.co_sanchit-gandhi_parler-tts-mini-v0.1-expresso-concatenated_run_parler_tts_training.py - {'id': 'QXJ0aWZhY3Q6ODM0NzI5Njgw', 'state': 'PENDING', 'artifactSequence': {'id': 'QXJ0aWZhY3RDb2xsZWN0aW9uOjE3NDIzMTI1Mw==', 'latestArtifact': None}}
+2024-05-13 20:50:37,629 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:50:37,629 INFO    SenderThread:1244488 [sender.py:send_request_defer():609] handle sender defer: 9
+2024-05-13 20:50:37,630 INFO    SenderThread:1244488 [dir_watcher.py:finish():358] shutting down directory watcher
+2024-05-13 20:50:38,597 INFO    SenderThread:1244488 [dir_watcher.py:finish():388] scan: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files
+2024-05-13 20:50:38,597 INFO    SenderThread:1244488 [dir_watcher.py:finish():402] scan save: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/conda-environment.yaml conda-environment.yaml
+2024-05-13 20:50:38,597 INFO    SenderThread:1244488 [dir_watcher.py:finish():402] scan save: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/wandb-summary.json wandb-summary.json
+2024-05-13 20:50:38,597 INFO    SenderThread:1244488 [dir_watcher.py:finish():402] scan save: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/output.log output.log
+2024-05-13 20:50:38,597 INFO    SenderThread:1244488 [dir_watcher.py:finish():402] scan save: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/requirements.txt requirements.txt
+2024-05-13 20:50:38,601 INFO    SenderThread:1244488 [dir_watcher.py:finish():402] scan save: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/config.yaml config.yaml
+2024-05-13 20:50:38,603 INFO    SenderThread:1244488 [dir_watcher.py:finish():402] scan save: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/wandb-metadata.json wandb-metadata.json
+2024-05-13 20:50:38,604 INFO    SenderThread:1244488 [sender.py:transition_state():613] send defer: 10
+2024-05-13 20:50:38,606 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: poll_exit
+2024-05-13 20:50:38,608 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:50:38,609 INFO    HandlerThread:1244488 [handler.py:handle_request_defer():184] handle defer: 10
+2024-05-13 20:50:38,609 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:50:38,609 INFO    SenderThread:1244488 [sender.py:send_request_defer():609] handle sender defer: 10
+2024-05-13 20:50:38,609 INFO    SenderThread:1244488 [file_pusher.py:finish():169] shutting down file pusher
+2024-05-13 20:50:38,936 INFO    wandb-upload_0:1244488 [upload_job.py:push():130] Uploaded file /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/wandb-summary.json
+2024-05-13 20:50:38,945 INFO    wandb-upload_1:1244488 [upload_job.py:push():130] Uploaded file /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/conda-environment.yaml
+2024-05-13 20:50:39,053 INFO    wandb-upload_3:1244488 [upload_job.py:push():130] Uploaded file /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/requirements.txt
+2024-05-13 20:50:39,066 INFO    wandb-upload_2:1244488 [upload_job.py:push():130] Uploaded file /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/output.log
+2024-05-13 20:50:39,072 INFO    wandb-upload_4:1244488 [upload_job.py:push():130] Uploaded file /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/files/config.yaml
+2024-05-13 20:50:39,136 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: poll_exit
+2024-05-13 20:50:39,136 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: poll_exit
+2024-05-13 20:50:39,272 INFO    Thread-11 (_thread_body):1244488 [sender.py:transition_state():613] send defer: 11
+2024-05-13 20:50:39,273 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:50:39,273 INFO    HandlerThread:1244488 [handler.py:handle_request_defer():184] handle defer: 11
+2024-05-13 20:50:39,273 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:50:39,273 INFO    SenderThread:1244488 [sender.py:send_request_defer():609] handle sender defer: 11
+2024-05-13 20:50:39,273 INFO    SenderThread:1244488 [file_pusher.py:join():175] waiting for file pusher
+2024-05-13 20:50:39,273 INFO    SenderThread:1244488 [sender.py:transition_state():613] send defer: 12
+2024-05-13 20:50:39,274 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:50:39,274 INFO    HandlerThread:1244488 [handler.py:handle_request_defer():184] handle defer: 12
+2024-05-13 20:50:39,274 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:50:39,274 INFO    SenderThread:1244488 [sender.py:send_request_defer():609] handle sender defer: 12
+2024-05-13 20:50:39,274 INFO    SenderThread:1244488 [file_stream.py:finish():601] file stream finish called
+2024-05-13 20:50:39,402 INFO    SenderThread:1244488 [file_stream.py:finish():605] file stream finish is done
+2024-05-13 20:50:39,402 INFO    SenderThread:1244488 [sender.py:transition_state():613] send defer: 13
+2024-05-13 20:50:39,402 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:50:39,402 INFO    HandlerThread:1244488 [handler.py:handle_request_defer():184] handle defer: 13
+2024-05-13 20:50:39,403 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:50:39,403 INFO    SenderThread:1244488 [sender.py:send_request_defer():609] handle sender defer: 13
+2024-05-13 20:50:39,403 INFO    SenderThread:1244488 [sender.py:transition_state():613] send defer: 14
+2024-05-13 20:50:39,403 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:50:39,403 INFO    HandlerThread:1244488 [handler.py:handle_request_defer():184] handle defer: 14
+2024-05-13 20:50:39,403 DEBUG   SenderThread:1244488 [sender.py:send():378] send: final
+2024-05-13 20:50:39,403 DEBUG   SenderThread:1244488 [sender.py:send():378] send: footer
+2024-05-13 20:50:39,403 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:50:39,403 INFO    SenderThread:1244488 [sender.py:send_request_defer():609] handle sender defer: 14
+2024-05-13 20:50:39,404 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: poll_exit
+2024-05-13 20:50:39,404 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: poll_exit
+2024-05-13 20:50:39,405 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: poll_exit
+2024-05-13 20:50:39,405 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: poll_exit
+2024-05-13 20:50:39,405 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: server_info
+2024-05-13 20:50:39,406 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: get_summary
+2024-05-13 20:50:39,406 DEBUG   SenderThread:1244488 [sender.py:send_request():405] send_request: server_info
+2024-05-13 20:50:39,408 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: sampled_history
+2024-05-13 20:50:39,408 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: internal_messages
+2024-05-13 20:50:39,538 INFO    MainThread:1244488 [wandb_run.py:_footer_history_summary_info():3994] rendering history
+2024-05-13 20:50:39,538 INFO    MainThread:1244488 [wandb_run.py:_footer_history_summary_info():4026] rendering summary
+2024-05-13 20:50:39,538 INFO    MainThread:1244488 [wandb_run.py:_footer_sync_info():3953] logging synced files
+2024-05-13 20:50:39,539 DEBUG   HandlerThread:1244488 [handler.py:handle_request():158] handle_request: shutdown
+2024-05-13 20:50:39,539 INFO    HandlerThread:1244488 [handler.py:finish():882] shutting down handler
+2024-05-13 20:50:40,406 INFO    WriterThread:1244488 [datastore.py:close():296] close: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/run-y7fy6vtp.wandb
+2024-05-13 20:50:40,538 INFO    SenderThread:1244488 [sender.py:finish():1545] shutting down sender
+2024-05-13 20:50:40,538 INFO    SenderThread:1244488 [file_pusher.py:finish():169] shutting down file pusher
+2024-05-13 20:50:40,538 INFO    SenderThread:1244488 [file_pusher.py:join():175] waiting for file pusher

wandb/run-20240513_204644-y7fy6vtp/logs/debug.log

@@ -0,0 +1,29 @@
+2024-05-13 20:46:44,143 INFO    MainThread:1244392 [wandb_setup.py:_flush():76] Current SDK version is 0.17.0
+2024-05-13 20:46:44,143 INFO    MainThread:1244392 [wandb_setup.py:_flush():76] Configure stats pid to 1244392
+2024-05-13 20:46:44,143 INFO    MainThread:1244392 [wandb_setup.py:_flush():76] Loading settings from /home/sanchit/.config/wandb/settings
+2024-05-13 20:46:44,143 INFO    MainThread:1244392 [wandb_setup.py:_flush():76] Loading settings from /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/settings
+2024-05-13 20:46:44,143 INFO    MainThread:1244392 [wandb_setup.py:_flush():76] Loading settings from environment variables: {}
+2024-05-13 20:46:44,143 INFO    MainThread:1244392 [wandb_setup.py:_flush():76] Applying setup settings: {'_disable_service': False}
+2024-05-13 20:46:44,143 INFO    MainThread:1244392 [wandb_setup.py:_flush():76] Inferring run settings from compute environment: {'program_relpath': 'run_parler_tts_training.py', 'program_abspath': '/raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/run_parler_tts_training.py', 'program': '/raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/./run_parler_tts_training.py'}
+2024-05-13 20:46:44,143 INFO    MainThread:1244392 [wandb_setup.py:_flush():76] Applying login settings: {}
+2024-05-13 20:46:44,143 INFO    MainThread:1244392 [wandb_init.py:_log_setup():520] Logging user logs to /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/logs/debug.log
+2024-05-13 20:46:44,143 INFO    MainThread:1244392 [wandb_init.py:_log_setup():521] Logging internal logs to /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_204644-y7fy6vtp/logs/debug-internal.log
+2024-05-13 20:46:44,143 INFO    MainThread:1244392 [wandb_init.py:init():560] calling init triggers
+2024-05-13 20:46:44,143 INFO    MainThread:1244392 [wandb_init.py:init():567] wandb.init called with sweep_config: {}
+config: {}
+2024-05-13 20:46:44,144 INFO    MainThread:1244392 [wandb_init.py:init():610] starting backend
+2024-05-13 20:46:44,144 INFO    MainThread:1244392 [wandb_init.py:init():614] setting up manager
+2024-05-13 20:46:44,147 INFO    MainThread:1244392 [backend.py:_multiprocessing_setup():105] multiprocessing start_methods=fork,spawn,forkserver, using: spawn
+2024-05-13 20:46:44,147 INFO    MainThread:1244392 [wandb_init.py:init():622] backend started and connected
+2024-05-13 20:46:44,149 INFO    MainThread:1244392 [wandb_init.py:init():711] updated telemetry
+2024-05-13 20:46:44,152 INFO    MainThread:1244392 [wandb_init.py:init():744] communicating run to backend with 90.0 second timeout
+2024-05-13 20:46:44,531 INFO    MainThread:1244392 [wandb_run.py:_on_init():2396] communicating current version
+2024-05-13 20:46:44,589 INFO    MainThread:1244392 [wandb_run.py:_on_init():2405] got version response 
+2024-05-13 20:46:44,590 INFO    MainThread:1244392 [wandb_init.py:init():795] starting run threads in backend
+2024-05-13 20:46:47,163 INFO    MainThread:1244392 [wandb_run.py:_console_start():2374] atexit reg
+2024-05-13 20:46:47,164 INFO    MainThread:1244392 [wandb_run.py:_redirect():2229] redirect: wrap_raw
+2024-05-13 20:46:47,164 INFO    MainThread:1244392 [wandb_run.py:_redirect():2294] Wrapping output streams.
+2024-05-13 20:46:47,164 INFO    MainThread:1244392 [wandb_run.py:_redirect():2319] Redirects installed.
+2024-05-13 20:46:47,165 INFO    MainThread:1244392 [wandb_init.py:init():838] run started, returning control to user process
+2024-05-13 20:46:47,165 INFO    MainThread:1244392 [wandb_run.py:_config_callback():1376] config_cb None None {'learning_rate': 8e-05, 'model_name_or_path': 'parler-tts/parler_tts_mini_v0.1', 'num_train_epochs': 8.0, 'gradient_accumulation_steps': 8, 'per_device_train_batch_size': 16, 'global_batch_size': 16, 'mixed_precision': 'bf16', 'lr_scheduler_type': 'SchedulerType.COSINE', 'warmup_steps': 250, 'freeze_text_encoder': True, 'max_duration_in_seconds': 30.0, 'weight_decay': 0.01, 'adam_beta1': 0.9, 'adam_beta2': 0.99, 'temperature': 1.0}
+2024-05-13 20:50:40,539 WARNING MsgRouterThr:1244392 [router.py:message_loop():77] message_loop has been closed

wandb/run-20240513_205248-d781ddha/files/conda-environment.yaml

@@ -0,0 +1,248 @@
+name: venv
+channels:
+  - defaults
+dependencies:
+  - _libgcc_mutex=0.1=main
+  - _openmp_mutex=5.1=1_gnu
+  - bzip2=1.0.8=h5eee18b_6
+  - ca-certificates=2024.3.11=h06a4308_0
+  - ld_impl_linux-64=2.38=h1181459_1
+  - libffi=3.4.4=h6a678d5_1
+  - libgcc-ng=11.2.0=h1234567_1
+  - libgomp=11.2.0=h1234567_1
+  - libstdcxx-ng=11.2.0=h1234567_1
+  - libuuid=1.41.5=h5eee18b_0
+  - ncurses=6.4=h6a678d5_0
+  - openssl=3.0.13=h7f8727e_1
+  - pip=24.0=py311h06a4308_0
+  - python=3.11.9=h955ad1f_0
+  - readline=8.2=h5eee18b_0
+  - setuptools=69.5.1=py311h06a4308_0
+  - sqlite=3.45.3=h5eee18b_0
+  - tk=8.6.14=h39e8969_0
+  - wheel=0.43.0=py311h06a4308_0
+  - xz=5.4.6=h5eee18b_1
+  - zlib=1.2.13=h5eee18b_1
+  - pip:
+      - absl-py==2.1.0
+      - accelerate==0.30.0
+      - aiohttp==3.9.5
+      - aiosignal==1.3.1
+      - aniso8601==9.0.1
+      - annotated-types==0.6.0
+      - anyio==4.3.0
+      - argbind==0.3.7
+      - argon2-cffi==23.1.0
+      - argon2-cffi-bindings==21.2.0
+      - arrow==1.3.0
+      - asttokens==2.4.1
+      - async-lru==2.0.4
+      - attrs==23.2.0
+      - audioread==3.0.1
+      - babel==2.15.0
+      - beautifulsoup4==4.12.3
+      - bidict==0.23.1
+      - bitsandbytes==0.43.1
+      - bleach==6.1.0
+      - certifi==2024.2.2
+      - cffi==1.16.0
+      - charset-normalizer==3.3.2
+      - click==8.1.7
+      - coloredlogs==14.0
+      - comm==0.2.2
+      - contourpy==1.2.1
+      - cycler==0.12.1
+      - datasets==2.19.1
+      - debugpy==1.8.1
+      - decorator==5.1.1
+      - defusedxml==0.7.1
+      - descript-audio-codec==1.0.0
+      - descript-audiotools==0.7.2
+      - dill==0.3.8
+      - dnspython==2.3.0
+      - docker-pycreds==0.4.0
+      - docstring-parser==0.16
+      - editdistance==0.8.1
+      - einops==0.8.0
+      - et-xmlfile==1.1.0
+      - evaluate==0.4.2
+      - eventlet==0.36.1
+      - executing==2.0.1
+      - fastjsonschema==2.19.1
+      - ffmpy==0.3.2
+      - filelock==3.14.0
+      - fire==0.6.0
+      - flask==2.2.5
+      - flask-cors==4.0.1
+      - flask-restful==0.3.10
+      - flask-socketio==5.3.6
+      - flask-talisman==1.1.0
+      - flatten-dict==0.4.2
+      - fonttools==4.51.0
+      - fqdn==1.5.1
+      - frozenlist==1.4.1
+      - fsspec==2024.3.1
+      - future==1.0.0
+      - g2p==2.0.0
+      - gitdb==4.0.11
+      - gitpython==3.1.43
+      - greenlet==3.0.3
+      - grpcio==1.63.0
+      - h11==0.14.0
+      - httpcore==1.0.5
+      - httpx==0.27.0
+      - huggingface-hub==0.23.0
+      - humanfriendly==10.0
+      - idna==3.7
+      - importlib-resources==6.4.0
+      - ipdb==0.13.13
+      - ipykernel==6.29.4
+      - ipython==8.24.0
+      - isoduration==20.11.0
+      - itsdangerous==2.2.0
+      - jedi==0.19.1
+      - jinja2==3.1.4
+      - jiwer==3.0.4
+      - joblib==1.4.2
+      - json5==0.9.25
+      - jsonpointer==2.4
+      - jsonschema==4.22.0
+      - jsonschema-specifications==2023.12.1
+      - julius==0.2.7
+      - jupyter-client==8.6.1
+      - jupyter-core==5.7.2
+      - jupyter-events==0.10.0
+      - jupyter-lsp==2.2.5
+      - jupyter-server==2.14.0
+      - jupyter-server-terminals==0.5.3
+      - jupyterlab==4.2.0
+      - jupyterlab-pygments==0.3.0
+      - jupyterlab-server==2.27.1
+      - kiwisolver==1.4.5
+      - lazy-loader==0.4
+      - librosa==0.10.2
+      - llvmlite==0.42.0
+      - markdown==3.6
+      - markdown-it-py==3.0.0
+      - markdown2==2.4.13
+      - markupsafe==2.1.5
+      - matplotlib==3.8.4
+      - matplotlib-inline==0.1.7
+      - mdurl==0.1.2
+      - mistune==3.0.2
+      - mpmath==1.3.0
+      - msgpack==1.0.8
+      - multidict==6.0.5
+      - multiprocess==0.70.16
+      - munkres==1.1.4
+      - nbclient==0.10.0
+      - nbconvert==7.16.4
+      - nbformat==5.10.4
+      - nest-asyncio==1.6.0
+      - networkx==3.3
+      - notebook-shim==0.2.4
+      - numba==0.59.1
+      - numpy==1.26.4
+      - nvidia-cublas-cu12==12.1.3.1
+      - nvidia-cuda-cupti-cu12==12.1.105
+      - nvidia-cuda-nvrtc-cu12==12.1.105
+      - nvidia-cuda-runtime-cu12==12.1.105
+      - nvidia-cudnn-cu12==8.9.2.26
+      - nvidia-cufft-cu12==11.0.2.54
+      - nvidia-curand-cu12==10.3.2.106
+      - nvidia-cusolver-cu12==11.4.5.107
+      - nvidia-cusparse-cu12==12.1.0.106
+      - nvidia-nccl-cu12==2.20.5
+      - nvidia-nvjitlink-cu12==12.4.127
+      - nvidia-nvtx-cu12==12.1.105
+      - openpyxl==3.1.2
+      - overrides==7.7.0
+      - packaging==24.0
+      - pandas==2.2.2
+      - pandocfilters==1.5.1
+      - panphon==0.20.0
+      - parler-tts==0.1
+      - parso==0.8.4
+      - pexpect==4.9.0
+      - pillow==10.3.0
+      - platformdirs==4.2.1
+      - pooch==1.8.1
+      - prometheus-client==0.20.0
+      - prompt-toolkit==3.0.43
+      - protobuf==3.19.6
+      - psutil==5.9.8
+      - ptyprocess==0.7.0
+      - pure-eval==0.2.2
+      - pyarrow==16.0.0
+      - pyarrow-hotfix==0.6
+      - pycparser==2.22
+      - pydantic==2.7.1
+      - pydantic-core==2.18.2
+      - pygments==2.18.0
+      - pyloudnorm==0.1.1
+      - pyparsing==3.1.2
+      - pystoi==0.4.1
+      - python-dateutil==2.9.0.post0
+      - python-engineio==4.9.0
+      - python-json-logger==2.0.7
+      - python-socketio==5.11.2
+      - pytz==2024.1
+      - pyyaml==6.0.1
+      - pyzmq==26.0.3
+      - randomname==0.2.1
+      - rapidfuzz==3.9.0
+      - referencing==0.35.1
+      - regex==2024.4.28
+      - requests==2.31.0
+      - rfc3339-validator==0.1.4
+      - rfc3986-validator==0.1.1
+      - rich==13.7.1
+      - rpds-py==0.18.1
+      - safetensors==0.4.3
+      - scikit-learn==1.4.2
+      - scipy==1.13.0
+      - send2trash==1.8.3
+      - sentencepiece==0.2.0
+      - sentry-sdk==2.1.1
+      - setproctitle==1.3.3
+      - simple-websocket==1.0.0
+      - six==1.16.0
+      - smmap==5.0.1
+      - sniffio==1.3.1
+      - soundfile==0.12.1
+      - soupsieve==2.5
+      - soxr==0.3.7
+      - stack-data==0.6.3
+      - sympy==1.12
+      - tensorboard==2.16.2
+      - tensorboard-data-server==0.7.2
+      - termcolor==2.4.0
+      - terminado==0.18.1
+      - text-unidecode==1.3
+      - threadpoolctl==3.5.0
+      - tinycss2==1.3.0
+      - tokenizers==0.19.1
+      - torch==2.3.0
+      - torch-stoi==0.2.1
+      - torchaudio==2.3.0
+      - tornado==6.4
+      - tqdm==4.66.4
+      - traitlets==5.14.3
+      - transformers==4.41.0.dev0
+      - triton==2.3.0
+      - types-python-dateutil==2.9.0.20240316
+      - typing-extensions==4.11.0
+      - tzdata==2024.1
+      - unicodecsv==0.14.1
+      - uri-template==1.3.0
+      - urllib3==2.2.1
+      - wandb==0.17.0
+      - wcwidth==0.2.13
+      - webcolors==1.13
+      - webencodings==0.5.1
+      - websocket-client==1.8.0
+      - werkzeug==3.0.3
+      - wsproto==1.2.0
+      - xxhash==3.4.1
+      - yarl==1.9.4
+prefix: /home/sanchit/miniconda3/envs/venv

wandb/run-20240513_205248-d781ddha/files/config.yaml

@@ -0,0 +1,86 @@
+wandb_version: 1
+
+_wandb:
+  desc: null
+  value:
+    python_version: 3.11.9
+    cli_version: 0.17.0
+    framework: huggingface
+    huggingface_version: 4.41.0.dev0
+    is_jupyter_run: false
+    is_kaggle_kernel: false
+    start_time: 1715626368
+    t:
+      1:
+      - 1
+      - 5
+      - 11
+      - 49
+      - 51
+      - 53
+      - 55
+      - 71
+      - 100
+      2:
+      - 1
+      - 5
+      - 11
+      - 49
+      - 51
+      - 53
+      - 55
+      - 71
+      - 100
+      3:
+      - 23
+      4: 3.11.9
+      5: 0.17.0
+      6: 4.41.0.dev0
+      8:
+      - 5
+      13: linux-x86_64
+learning_rate:
+  desc: null
+  value: 8.0e-05
+model_name_or_path:
+  desc: null
+  value: parler-tts/parler_tts_mini_v0.1
+num_train_epochs:
+  desc: null
+  value: 8.0
+gradient_accumulation_steps:
+  desc: null
+  value: 8
+per_device_train_batch_size:
+  desc: null
+  value: 16
+global_batch_size:
+  desc: null
+  value: 16
+mixed_precision:
+  desc: null
+  value: bf16
+lr_scheduler_type:
+  desc: null
+  value: SchedulerType.COSINE
+warmup_steps:
+  desc: null
+  value: 250
+freeze_text_encoder:
+  desc: null
+  value: true
+max_duration_in_seconds:
+  desc: null
+  value: 30.0
+weight_decay:
+  desc: null
+  value: 0.01
+adam_beta1:
+  desc: null
+  value: 0.9
+adam_beta2:
+  desc: null
+  value: 0.99
+temperature:
+  desc: null
+  value: 1.0

wandb/run-20240513_205248-d781ddha/files/output.log

@@ -0,0 +1,576 @@
+05/13/2024 20:52:51 - WARNING - __main__ - Process rank: 0, device: cuda:0, n_gpu: 1, distributed training: False, 16-bits training: False
+05/13/2024 20:52:51 - INFO - __main__ - Training/evaluation parameters ParlerTTSTrainingArguments(
+_n_gpu=1,
+accelerator_config={'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None},
+adafactor=False,
+adam_beta1=0.9,
+adam_beta2=0.99,
+adam_epsilon=1e-08,
+audio_encoder_per_device_batch_size=4,
+auto_find_batch_size=False,
+batch_eval_metrics=False,
+bf16=False,
+bf16_full_eval=False,
+data_seed=None,
+dataloader_drop_last=False,
+dataloader_num_workers=4,
+dataloader_persistent_workers=False,
+dataloader_pin_memory=True,
+dataloader_prefetch_factor=None,
+ddp_backend=None,
+ddp_broadcast_buffers=None,
+ddp_bucket_cap_mb=None,
+ddp_find_unused_parameters=None,
+ddp_timeout=1800,
+debug=[],
+deepspeed=None,
+disable_tqdm=False,
+dispatch_batches=None,
+do_eval=True,
+do_predict=False,
+do_train=True,
+dtype=bfloat16,
+eval_accumulation_steps=None,
+eval_delay=0,
+eval_do_concat_batches=True,
+eval_steps=None,
+eval_strategy=IntervalStrategy.EPOCH,
+evaluation_strategy=epoch,
+fp16=False,
+fp16_backend=auto,
+fp16_full_eval=False,
+fp16_opt_level=O1,
+fsdp=[],
+fsdp_config={'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False},
+fsdp_min_num_params=0,
+fsdp_transformer_layer_cls_to_wrap=None,
+full_determinism=False,
+generation_config=None,
+generation_max_length=None,
+generation_num_beams=None,
+gradient_accumulation_steps=8,
+gradient_checkpointing=True,
+gradient_checkpointing_kwargs=None,
+greater_is_better=None,
+group_by_length=True,
+half_precision_backend=auto,
+hub_always_push=False,
+hub_model_id=None,
+hub_private_repo=False,
+hub_strategy=HubStrategy.EVERY_SAVE,
+hub_token=<HUB_TOKEN>,
+ignore_data_skip=False,
+include_inputs_for_metrics=True,
+include_num_input_tokens_seen=False,
+include_tokens_per_second=False,
+jit_mode_eval=False,
+label_names=None,
+label_smoothing_factor=0.0,
+learning_rate=8e-05,
+length_column_name=length,
+load_best_model_at_end=False,
+local_rank=0,
+log_level=passive,
+log_level_replica=warning,
+log_on_each_node=True,
+logging_dir=../output_dir_training_concat/runs/May13_20-52-47_hf-dgx-01,
+logging_first_step=False,
+logging_nan_inf_filter=True,
+logging_steps=2,
+logging_strategy=IntervalStrategy.STEPS,
+lr_scheduler_kwargs={},
+lr_scheduler_type=SchedulerType.COSINE,
+max_grad_norm=1.0,
+max_steps=-1,
+metric_for_best_model=None,
+mp_parameters=,
+neftune_noise_alpha=None,
+no_cuda=False,
+num_train_epochs=8.0,
+optim=OptimizerNames.ADAMW_TORCH,
+optim_args=None,
+optim_target_modules=None,
+output_dir=../output_dir_training_concat/,
+overwrite_output_dir=True,
+past_index=-1,
+per_device_eval_batch_size=16,
+per_device_train_batch_size=16,
+predict_with_generate=True,
+prediction_loss_only=False,
+push_to_hub=False,
+push_to_hub_model_id=None,
+push_to_hub_organization=None,
+push_to_hub_token=<PUSH_TO_HUB_TOKEN>,
+ray_scope=last,
+remove_unused_columns=True,
+report_to=['wandb'],
+restore_callback_states_from_checkpoint=False,
+resume_from_checkpoint=None,
+run_name=../output_dir_training_concat/,
+save_on_each_node=False,
+save_only_model=False,
+save_safetensors=True,
+save_steps=72,
+save_strategy=IntervalStrategy.STEPS,
+save_total_limit=5,
+seed=456,
+skip_memory_metrics=True,
+sortish_sampler=False,
+split_batches=None,
+tf32=None,
+torch_compile=False,
+torch_compile_backend=None,
+torch_compile_mode=None,
+torchdynamo=None,
+tpu_metrics_debug=False,
+tpu_num_cores=None,
+use_cpu=False,
+use_ipex=False,
+use_legacy_prediction_loop=False,
+use_mps_device=False,
+warmup_ratio=0.0,
+warmup_steps=250,
+weight_decay=0.01,
+)
+05/13/2024 20:52:53 - WARNING - __main__ - Disabling fast tokenizer warning: https://github.com/huggingface/transformers/blob/main/src/transformers/tokenization_utils_base.py#L3231-L3235
+loading configuration file preprocessor_config.json from cache at /raid/.cache/huggingface/models--parler-tts--dac_44khZ_8kbps/snapshots/db52bea859d9411e0beb44a3ea923a8731ee4197/preprocessor_config.json
+Feature extractor EncodecFeatureExtractor {
+  "chunk_length_s": null,
+  "feature_extractor_type": "EncodecFeatureExtractor",
+  "feature_size": 1,
+  "overlap": null,
+  "padding_side": "right",
+  "padding_value": 0.0,
+  "return_attention_mask": true,
+  "sampling_rate": 44100
+}
+loading file spiece.model from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/spiece.model
+loading file tokenizer.json from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/tokenizer.json
+loading file added_tokens.json from cache at None
+loading file special_tokens_map.json from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/special_tokens_map.json
+loading file tokenizer_config.json from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/tokenizer_config.json
+You set `add_prefix_space`. The tokenizer needs to be converted from the slow tokenizers
+loading file spiece.model from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/spiece.model
+loading file tokenizer.json from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/tokenizer.json
+loading file added_tokens.json from cache at None
+loading file special_tokens_map.json from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/special_tokens_map.json
+loading file tokenizer_config.json from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/tokenizer_config.json
+Combining datasets...:   0%|                                                                      | 0/1 [00:00<?, ?it/s]
+Downloading readme: 100%|██████████████████████████████████████████████████████████████| 487/487 [00:00<00:00, 2.51MB/s]
+Downloading readme:   0%|                                                                     | 0.00/487 [00:00<?, ?B/s]
+Downloading data:   0%|                                                                      | 0.00/529k [00:00<?, ?B/s]
+05/13/2024 20:52:57 - WARNING - datasets.builder - Setting num_proc from 2 back to 1 for the train split to disable multiprocessing as it only contains one shard.
+Setting num_proc from 2 back to 1 for the train split to disable multiprocessing as it only contains one shard.1.53MB/s]
+Generating train split: 100%|████████████████████████████████████████████| 5347/5347 [00:00<00:00, 530764.98 examples/s]
+Combining datasets...: 100%|██████████████████████████████████████████████████████████████| 1/1 [00:04<00:00,  4.79s/it]
+Combining datasets...:   0%|                                                                      | 0/1 [00:00<?, ?it/s]
+05/13/2024 20:52:59 - INFO - __main__ - Merging sanchit-gandhi/expresso-concatenated-half-normal - train with sanchit-gandhi/expresso-concatenated-half-normal-tags-mistral - train
+Combining datasets...: 100%|██████████████████████████████████████████████████████████████| 1/1 [00:03<00:00,  3.31s/it]
+/home/sanchit/miniconda3/envs/venv/lib/python3.11/site-packages/huggingface_hub/file_download.py:1132: FutureWarning: `resume_download` is deprecated and will be removed in version 1.0.0. Downloads always resume when possible. If you want to force a new download, use `force_download=True`.
+  warnings.warn(
+loading configuration file config.json from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/config.json
+Model config ParlerTTSConfig {
+  "_name_or_path": "/fsx/yoach/tmp/artefacts/training-400M-punctuated-v2/",
+  "architectures": [
+    "ParlerTTSForConditionalGeneration"
+  ],
+  "audio_encoder": {
+    "_name_or_path": "ylacombe/dac_44khZ_8kbps",
+    "add_cross_attention": false,
+    "architectures": [
+      "DACModel"
+    ],
+    "bad_words_ids": null,
+    "begin_suppress_tokens": null,
+    "bos_token_id": null,
+    "chunk_size_feed_forward": 0,
+    "codebook_size": 1024,
+    "cross_attention_hidden_size": null,
+    "decoder_start_token_id": null,
+    "diversity_penalty": 0.0,
+    "do_sample": false,
+    "early_stopping": false,
+    "encoder_no_repeat_ngram_size": 0,
+    "eos_token_id": null,
+    "exponential_decay_length_penalty": null,
+    "finetuning_task": null,
+    "forced_bos_token_id": null,
+    "forced_eos_token_id": null,
+    "frame_rate": 86,
+    "id2label": {
+      "0": "LABEL_0",
+      "1": "LABEL_1"
+    },
+    "is_decoder": false,
+    "is_encoder_decoder": false,
+    "label2id": {
+      "LABEL_0": 0,
+      "LABEL_1": 1
+    },
+    "latent_dim": 1024,
+    "length_penalty": 1.0,
+    "max_length": 20,
+    "min_length": 0,
+    "model_bitrate": 8,
+    "model_type": "dac",
+    "no_repeat_ngram_size": 0,
+    "num_beam_groups": 1,
+    "num_beams": 1,
+    "num_codebooks": 9,
+    "num_return_sequences": 1,
+    "output_attentions": false,
+    "output_hidden_states": false,
+    "output_scores": false,
+    "pad_token_id": null,
+    "prefix": null,
+    "problem_type": null,
+    "pruned_heads": {},
+    "remove_invalid_values": false,
+    "repetition_penalty": 1.0,
+    "return_dict": true,
+    "return_dict_in_generate": false,
+    "sampling_rate": 44100,
+    "sep_token_id": null,
+    "suppress_tokens": null,
+    "task_specific_params": null,
+    "temperature": 1.0,
+    "tf_legacy_loss": false,
+    "tie_encoder_decoder": false,
+    "tie_word_embeddings": true,
+    "tokenizer_class": null,
+    "top_k": 50,
+    "top_p": 1.0,
+    "torch_dtype": "float32",
+    "torchscript": false,
+    "typical_p": 1.0,
+    "use_bfloat16": false
+  },
+  "decoder": {
+    "_name_or_path": "/fsx/yoach/tmp/artefacts/decoder_400M/",
+    "activation_dropout": 0.0,
+    "activation_function": "gelu",
+    "add_cross_attention": true,
+    "architectures": [
+      "ParlerTTSForCausalLM"
+    ],
+    "attention_dropout": 0.0,
+    "bad_words_ids": null,
+    "begin_suppress_tokens": null,
+    "bos_token_id": 1025,
+    "chunk_size_feed_forward": 0,
+    "cross_attention_hidden_size": null,
+    "decoder_start_token_id": null,
+    "diversity_penalty": 0.0,
+    "do_sample": false,
+    "dropout": 0.1,
+    "early_stopping": false,
+    "encoder_no_repeat_ngram_size": 0,
+    "eos_token_id": 1024,
+    "exponential_decay_length_penalty": null,
+    "ffn_dim": 4096,
+    "finetuning_task": null,
+    "forced_bos_token_id": null,
+    "forced_eos_token_id": null,
+    "hidden_size": 1024,
+    "id2label": {
+      "0": "LABEL_0",
+      "1": "LABEL_1"
+    },
+    "initializer_factor": 0.02,
+    "is_decoder": true,
+    "is_encoder_decoder": false,
+    "label2id": {
+      "LABEL_0": 0,
+      "LABEL_1": 1
+    },
+    "layerdrop": 0.0,
+    "length_penalty": 1.0,
+    "max_length": 20,
+    "max_position_embeddings": 4096,
+    "min_length": 0,
+    "model_type": "parler_tts_decoder",
+    "no_repeat_ngram_size": 0,
+    "num_attention_heads": 16,
+    "num_beam_groups": 1,
+    "num_beams": 1,
+    "num_codebooks": 9,
+    "num_hidden_layers": 24,
+    "num_return_sequences": 1,
+    "output_attentions": false,
+    "output_hidden_states": false,
+    "output_scores": false,
+    "pad_token_id": 1024,
+    "prefix": null,
+    "problem_type": null,
+    "pruned_heads": {},
+    "remove_invalid_values": false,
+    "repetition_penalty": 1.0,
+    "return_dict": true,
+    "return_dict_in_generate": false,
+    "scale_embedding": false,
+    "sep_token_id": null,
+    "suppress_tokens": null,
+    "task_specific_params": null,
+    "temperature": 1.0,
+    "tf_legacy_loss": false,
+    "tie_encoder_decoder": false,
+    "tie_word_embeddings": false,
+    "tokenizer_class": null,
+    "top_k": 50,
+    "top_p": 1.0,
+    "torch_dtype": "float32",
+    "torchscript": false,
+    "typical_p": 1.0,
+    "use_bfloat16": false,
+    "use_cache": true,
+    "vocab_size": 1088
+  },
+  "decoder_start_token_id": 1025,
+  "is_encoder_decoder": true,
+  "model_type": "parler_tts",
+  "pad_token_id": 1024,
+  "text_encoder": {
+    "_name_or_path": "google/flan-t5-base",
+    "add_cross_attention": false,
+    "architectures": [
+      "T5ForConditionalGeneration"
+    ],
+    "bad_words_ids": null,
+    "begin_suppress_tokens": null,
+    "bos_token_id": null,
+    "chunk_size_feed_forward": 0,
+    "classifier_dropout": 0.0,
+    "cross_attention_hidden_size": null,
+    "d_ff": 2048,
+    "d_kv": 64,
+    "d_model": 768,
+    "decoder_start_token_id": 0,
+    "dense_act_fn": "gelu_new",
+    "diversity_penalty": 0.0,
+    "do_sample": false,
+    "dropout_rate": 0.1,
+    "early_stopping": false,
+    "encoder_no_repeat_ngram_size": 0,
+    "eos_token_id": 1,
+    "exponential_decay_length_penalty": null,
+    "feed_forward_proj": "gated-gelu",
+    "finetuning_task": null,
+    "forced_bos_token_id": null,
+    "forced_eos_token_id": null,
+    "id2label": {
+      "0": "LABEL_0",
+      "1": "LABEL_1"
+    },
+    "initializer_factor": 1.0,
+    "is_decoder": false,
+    "is_encoder_decoder": true,
+    "is_gated_act": true,
+    "label2id": {
+      "LABEL_0": 0,
+      "LABEL_1": 1
+    },
+    "layer_norm_epsilon": 1e-06,
+    "length_penalty": 1.0,
+    "max_length": 20,
+    "min_length": 0,
+    "model_type": "t5",
+    "n_positions": 512,
+    "no_repeat_ngram_size": 0,
+    "num_beam_groups": 1,
+    "num_beams": 1,
+    "num_decoder_layers": 12,
+    "num_heads": 12,
+    "num_layers": 12,
+    "num_return_sequences": 1,
+    "output_attentions": false,
+    "output_hidden_states": false,
+    "output_past": true,
+    "output_scores": false,
+    "pad_token_id": 0,
+    "prefix": null,
+    "problem_type": null,
+    "pruned_heads": {},
+    "relative_attention_max_distance": 128,
+    "relative_attention_num_buckets": 32,
+    "remove_invalid_values": false,
+    "repetition_penalty": 1.0,
+    "return_dict": true,
+    "return_dict_in_generate": false,
+    "sep_token_id": null,
+    "suppress_tokens": null,
+    "task_specific_params": {
+      "summarization": {
+        "early_stopping": true,
+        "length_penalty": 2.0,
+        "max_length": 200,
+        "min_length": 30,
+        "no_repeat_ngram_size": 3,
+        "num_beams": 4,
+        "prefix": "summarize: "
+      },
+      "translation_en_to_de": {
+        "early_stopping": true,
+        "max_length": 300,
+        "num_beams": 4,
+        "prefix": "translate English to German: "
+      },
+      "translation_en_to_fr": {
+        "early_stopping": true,
+        "max_length": 300,
+        "num_beams": 4,
+        "prefix": "translate English to French: "
+      },
+      "translation_en_to_ro": {
+        "early_stopping": true,
+        "max_length": 300,
+        "num_beams": 4,
+        "prefix": "translate English to Romanian: "
+      }
+    },
+    "temperature": 1.0,
+    "tf_legacy_loss": false,
+    "tie_encoder_decoder": false,
+    "tie_word_embeddings": false,
+    "tokenizer_class": null,
+    "top_k": 50,
+    "top_p": 1.0,
+    "torch_dtype": null,
+    "torchscript": false,
+    "typical_p": 1.0,
+    "use_bfloat16": false,
+    "use_cache": true,
+    "vocab_size": 32128
+  },
+  "torch_dtype": "float32",
+  "transformers_version": "4.41.0.dev0",
+  "vocab_size": 32128
+}
+loading weights file model.safetensors from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/model.safetensors
+Generate config GenerationConfig {
+  "decoder_start_token_id": 1025,
+  "pad_token_id": 1024
+}
+/home/sanchit/miniconda3/envs/venv/lib/python3.11/site-packages/torch/nn/utils/weight_norm.py:28: UserWarning: torch.nn.utils.weight_norm is deprecated in favor of torch.nn.utils.parametrizations.weight_norm.
+  warnings.warn("torch.nn.utils.weight_norm is deprecated in favor of torch.nn.utils.parametrizations.weight_norm.")
+Generate config GenerationConfig {
+  "bos_token_id": 1025,
+  "eos_token_id": 1024,
+  "pad_token_id": 1024
+}
+All model checkpoint weights were used when initializing ParlerTTSForConditionalGeneration.
+All the weights of ParlerTTSForConditionalGeneration were initialized from the model checkpoint at parler-tts/parler_tts_mini_v0.1.
+If your task is similar to the task the model of the checkpoint was trained on, you can already use ParlerTTSForConditionalGeneration for predictions without further training.
+loading configuration file generation_config.json from cache at /raid/.cache/huggingface/models--parler-tts--parler_tts_mini_v0.1/snapshots/e02fd18e77d38b49a85c7a9a85189a64b8472544/generation_config.json
+Generate config GenerationConfig {
+  "bos_token_id": 1025,
+  "decoder_start_token_id": 1025,
+  "do_sample": true,
+  "eos_token_id": 1024,
+  "guidance_scale": 1.0,
+  "max_length": 2580,
+  "min_new_tokens": 50,
+  "pad_token_id": 1024
+}
+gathered_tensor tensor([0], device='cuda:0')
+
+Filter (num_proc=2): 100%|██████████████████████████████████████████████████| 5347/5347 [00:05<00:00, 898.61 examples/s]
+
+Filter (num_proc=2): 100%|█████████████████████████████████████████████████████████| 8/8 [00:06<00:00,  1.22 examples/s]
+
+preprocess datasets (num_proc=2): 100%|█████████████████████████████████████| 5341/5341 [00:06<00:00, 766.49 examples/s]
+
+preprocess datasets (num_proc=2):  50%|██████████████████████                      | 4/8 [00:05<00:05,  1.27s/ examples]
+preprocess datasets (num_proc=2): 100%|████████████████████████████████████████████| 8/8 [00:06<00:00,  1.30 examples/s]
+  0%|                                                                                          | 0/1336 [00:00<?, ?it/s]/home/sanchit/miniconda3/envs/venv/lib/python3.11/site-packages/torch/nn/modules/conv.py:306: UserWarning: Plan failed with a cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_DESCRIPTOR: cudnnFinalize Descriptor Failed cudnn_status: CUDNN_STATUS_NOT_SUPPORTED (Triggered internally at ../aten/src/ATen/native/cudnn/Conv_v8.cpp:919.)
+  return F.conv1d(input, weight, bias, self.stride,
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+100%|██████████████████████████████████████████████████████████████████████████████▉| 1335/1336 [02:49<00:00,  7.90it/s]
+Traceback (most recent call last):
+  File "/raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/./run_parler_tts_training.py", line 1763, in <module>
+    main()
+  File "/raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/./run_parler_tts_training.py", line 1153, in main
+    lab = [l[:, : int(ratio * length)] for (l, ratio, length) in zip(lab, rat, lens)]
+                                                                 ^^^^^^^^^^^^^^^^^^^
+  File "/home/sanchit/miniconda3/envs/venv/lib/python3.11/site-packages/torch/_tensor.py", line 1047, in __iter__
+    raise TypeError("iteration over a 0-d tensor")
+TypeError: iteration over a 0-d tensor

wandb/run-20240513_205248-d781ddha/files/requirements.txt

@@ -0,0 +1,225 @@
+Babel==2.15.0
+Flask-Cors==4.0.1
+Flask-RESTful==0.3.10
+Flask-SocketIO==5.3.6
+Flask==2.2.5
+GitPython==3.1.43
+Jinja2==3.1.4
+Markdown==3.6
+MarkupSafe==2.1.5
+PyYAML==6.0.1
+Pygments==2.18.0
+Send2Trash==1.8.3
+Werkzeug==3.0.3
+absl-py==2.1.0
+accelerate==0.30.0
+aiohttp==3.9.5
+aiosignal==1.3.1
+aniso8601==9.0.1
+annotated-types==0.6.0
+anyio==4.3.0
+argbind==0.3.7
+argon2-cffi-bindings==21.2.0
+argon2-cffi==23.1.0
+arrow==1.3.0
+asttokens==2.4.1
+async-lru==2.0.4
+attrs==23.2.0
+audioread==3.0.1
+beautifulsoup4==4.12.3
+bidict==0.23.1
+bitsandbytes==0.43.1
+bleach==6.1.0
+certifi==2024.2.2
+cffi==1.16.0
+charset-normalizer==3.3.2
+click==8.1.7
+coloredlogs==14.0
+comm==0.2.2
+contourpy==1.2.1
+cycler==0.12.1
+datasets==2.19.1
+debugpy==1.8.1
+decorator==5.1.1
+defusedxml==0.7.1
+descript-audio-codec==1.0.0
+descript-audiotools==0.7.2
+dill==0.3.8
+dnspython==2.3.0
+docker-pycreds==0.4.0
+docstring_parser==0.16
+editdistance==0.8.1
+einops==0.8.0
+et-xmlfile==1.1.0
+evaluate==0.4.2
+eventlet==0.36.1
+executing==2.0.1
+fastjsonschema==2.19.1
+ffmpy==0.3.2
+filelock==3.14.0
+fire==0.6.0
+flask-talisman==1.1.0
+flatten-dict==0.4.2
+fonttools==4.51.0
+fqdn==1.5.1
+frozenlist==1.4.1
+fsspec==2024.3.1
+future==1.0.0
+g2p==2.0.0
+gitdb==4.0.11
+greenlet==3.0.3
+grpcio==1.63.0
+h11==0.14.0
+httpcore==1.0.5
+httpx==0.27.0
+huggingface-hub==0.23.0
+humanfriendly==10.0
+idna==3.7
+importlib_resources==6.4.0
+ipdb==0.13.13
+ipykernel==6.29.4
+ipython==8.24.0
+isoduration==20.11.0
+itsdangerous==2.2.0
+jedi==0.19.1
+jiwer==3.0.4
+joblib==1.4.2
+json5==0.9.25
+jsonpointer==2.4
+jsonschema-specifications==2023.12.1
+jsonschema==4.22.0
+julius==0.2.7
+jupyter-events==0.10.0
+jupyter-lsp==2.2.5
+jupyter_client==8.6.1
+jupyter_core==5.7.2
+jupyter_server==2.14.0
+jupyter_server_terminals==0.5.3
+jupyterlab==4.2.0
+jupyterlab_pygments==0.3.0
+jupyterlab_server==2.27.1
+kiwisolver==1.4.5
+lazy_loader==0.4
+librosa==0.10.2
+llvmlite==0.42.0
+markdown-it-py==3.0.0
+markdown2==2.4.13
+matplotlib-inline==0.1.7
+matplotlib==3.8.4
+mdurl==0.1.2
+mistune==3.0.2
+mpmath==1.3.0
+msgpack==1.0.8
+multidict==6.0.5
+multiprocess==0.70.16
+munkres==1.1.4
+nbclient==0.10.0
+nbconvert==7.16.4
+nbformat==5.10.4
+nest-asyncio==1.6.0
+networkx==3.3
+notebook_shim==0.2.4
+numba==0.59.1
+numpy==1.26.4
+nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cudnn-cu12==8.9.2.26
+nvidia-cufft-cu12==11.0.2.54
+nvidia-curand-cu12==10.3.2.106
+nvidia-cusolver-cu12==11.4.5.107
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-nccl-cu12==2.20.5
+nvidia-nvjitlink-cu12==12.4.127
+nvidia-nvtx-cu12==12.1.105
+openpyxl==3.1.2
+overrides==7.7.0
+packaging==24.0
+pandas==2.2.2
+pandocfilters==1.5.1
+panphon==0.20.0
+parler_tts==0.1
+parso==0.8.4
+pexpect==4.9.0
+pillow==10.3.0
+pip==24.0
+platformdirs==4.2.1
+pooch==1.8.1
+prometheus_client==0.20.0
+prompt-toolkit==3.0.43
+protobuf==3.19.6
+psutil==5.9.8
+ptyprocess==0.7.0
+pure-eval==0.2.2
+pyarrow-hotfix==0.6
+pyarrow==16.0.0
+pycparser==2.22
+pydantic==2.7.1
+pydantic_core==2.18.2
+pyloudnorm==0.1.1
+pyparsing==3.1.2
+pystoi==0.4.1
+python-dateutil==2.9.0.post0
+python-engineio==4.9.0
+python-json-logger==2.0.7
+python-socketio==5.11.2
+pytz==2024.1
+pyzmq==26.0.3
+randomname==0.2.1
+rapidfuzz==3.9.0
+referencing==0.35.1
+regex==2024.4.28
+requests==2.31.0
+rfc3339-validator==0.1.4
+rfc3986-validator==0.1.1
+rich==13.7.1
+rpds-py==0.18.1
+safetensors==0.4.3
+scikit-learn==1.4.2
+scipy==1.13.0
+sentencepiece==0.2.0
+sentry-sdk==2.1.1
+setproctitle==1.3.3
+setuptools==69.5.1
+simple-websocket==1.0.0
+six==1.16.0
+smmap==5.0.1
+sniffio==1.3.1
+soundfile==0.12.1
+soupsieve==2.5
+soxr==0.3.7
+stack-data==0.6.3
+sympy==1.12
+tensorboard-data-server==0.7.2
+tensorboard==2.16.2
+termcolor==2.4.0
+terminado==0.18.1
+text-unidecode==1.3
+threadpoolctl==3.5.0
+tinycss2==1.3.0
+tokenizers==0.19.1
+torch-stoi==0.2.1
+torch==2.3.0
+torchaudio==2.3.0
+tornado==6.4
+tqdm==4.66.4
+traitlets==5.14.3
+transformers==4.41.0.dev0
+transformers==4.41.0.dev0
+triton==2.3.0
+types-python-dateutil==2.9.0.20240316
+typing_extensions==4.11.0
+tzdata==2024.1
+unicodecsv==0.14.1
+uri-template==1.3.0
+urllib3==2.2.1
+wandb==0.17.0
+wcwidth==0.2.13
+webcolors==1.13
+webencodings==0.5.1
+websocket-client==1.8.0
+wheel==0.43.0
+wsproto==1.2.0
+xxhash==3.4.1
+yarl==1.9.4

wandb/run-20240513_205248-d781ddha/files/wandb-metadata.json

@@ -0,0 +1,804 @@
+{
+    "os": "Linux-5.4.0-166-generic-x86_64-with-glibc2.31",
+    "python": "3.11.9",
+    "heartbeatAt": "2024-05-13T18:52:49.375376",
+    "startedAt": "2024-05-13T18:52:48.864820",
+    "docker": null,
+    "cuda": null,
+    "args": [
+        "--model_name_or_path",
+        "parler-tts/parler_tts_mini_v0.1",
+        "--feature_extractor_name",
+        "parler-tts/dac_44khZ_8kbps",
+        "--description_tokenizer_name",
+        "parler-tts/parler_tts_mini_v0.1",
+        "--prompt_tokenizer_name",
+        "parler-tts/parler_tts_mini_v0.1",
+        "--report_to",
+        "wandb",
+        "--overwrite_output_dir",
+        "true",
+        "--train_dataset_name",
+        "sanchit-gandhi/expresso-concatenated-half-normal",
+        "--train_metadata_dataset_name",
+        "sanchit-gandhi/expresso-concatenated-half-normal-tags-mistral",
+        "--train_dataset_config_name",
+        "read",
+        "--train_split_name",
+        "train",
+        "--eval_dataset_name",
+        "sanchit-gandhi/expresso-concatenated-half-normal",
+        "--eval_metadata_dataset_name",
+        "sanchit-gandhi/expresso-concatenated-half-normal-tags-mistral",
+        "--eval_dataset_config_name",
+        "read",
+        "--eval_split_name",
+        "train",
+        "--max_eval_samples",
+        "8",
+        "--per_device_eval_batch_size",
+        "16",
+        "--target_audio_column_name",
+        "audio",
+        "--description_column_name",
+        "text_description",
+        "--prompt_column_name",
+        "text",
+        "--max_duration_in_seconds",
+        "30.0",
+        "--min_duration_in_seconds",
+        "2.0",
+        "--max_text_length",
+        "400",
+        "--preprocessing_num_workers",
+        "2",
+        "--do_train",
+        "true",
+        "--num_train_epochs",
+        "8",
+        "--gradient_accumulation_steps",
+        "8",
+        "--gradient_checkpointing",
+        "true",
+        "--per_device_train_batch_size",
+        "16",
+        "--learning_rate",
+        "0.00008",
+        "--adam_beta1",
+        "0.9",
+        "--adam_beta2",
+        "0.99",
+        "--weight_decay",
+        "0.01",
+        "--lr_scheduler_type",
+        "cosine",
+        "--warmup_steps",
+        "250",
+        "--logging_steps",
+        "2",
+        "--freeze_text_encoder",
+        "true",
+        "--audio_encoder_per_device_batch_size",
+        "4",
+        "--dtype",
+        "bfloat16",
+        "--seed",
+        "456",
+        "--output_dir",
+        "../output_dir_training_concat/",
+        "--temporary_save_to_disk",
+        "../audio_code_tmp_concat/",
+        "--save_to_disk",
+        "../tmp_dataset_audio_concat/",
+        "--dataloader_num_workers",
+        "4",
+        "--do_eval",
+        "--predict_with_generate",
+        "--include_inputs_for_metrics",
+        "--save_strategy",
+        "steps",
+        "--save_steps",
+        "72",
+        "--evaluation_strategy",
+        "epoch",
+        "--save_total_limit",
+        "5",
+        "--group_by_length",
+        "true"
+    ],
+    "state": "running",
+    "program": "/raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/./run_parler_tts_training.py",
+    "codePathLocal": "run_parler_tts_training.py",
+    "codePath": "run_parler_tts_training.py",
+    "git": {
+        "remote": "https://huggingface.co/sanchit-gandhi/parler-tts-mini-v0.1-expresso-concatenated",
+        "commit": "99f75adb5e13ee0ad87cce8deb3e71adc370b10e"
+    },
+    "email": "sanchit@huggingface.co",
+    "root": "/raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated",
+    "host": "hf-dgx-01",
+    "username": "sanchit",
+    "executable": "/home/sanchit/miniconda3/envs/venv/bin/python",
+    "cpu_count": 64,
+    "cpu_count_logical": 128,
+    "cpu_freq": {
+        "current": 2187.8050625000005,
+        "min": 1500.0,
+        "max": 2250.0
+    },
+    "cpu_freq_per_core": [
+        {
+            "current": 1713.071,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1712.469,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2280.258,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1717.303,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3312.578,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3051.416,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1658.955,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1663.383,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3332.084,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1663.447,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1663.494,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1665.248,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1793.968,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1794.861,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1793.472,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1792.28,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3341.098,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1659.571,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2090.509,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3330.01,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1661.972,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1661.373,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2744.439,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3341.302,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1931.367,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1663.659,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3325.735,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1663.731,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1715.888,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1713.973,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2361.772,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1717.358,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1716.521,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2341.51,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1715.513,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1711.953,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1793.443,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1796.101,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1794.087,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1794.423,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1790.765,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1796.312,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1792.734,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1794.642,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1659.721,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1663.181,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3315.724,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1660.427,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2966.59,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1668.725,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3300.161,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1671.705,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1661.312,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3339.377,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3333.308,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1662.747,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1660.781,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1660.176,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1664.39,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3336.046,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1793.809,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1794.079,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1594.389,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1622.233,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2414.464,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2429.082,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2278.496,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2421.333,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3330.367,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2412.212,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3102.925,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2356.416,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3312.565,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3044.183,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3296.93,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3060.353,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2411.85,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2433.228,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2464.094,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2438.304,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3306.628,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3318.093,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2748.09,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3318.082,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3316.036,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3318.945,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3313.086,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3340.023,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3306.8,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2153.453,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2690.778,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2155.725,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2149.357,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2147.249,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2359.311,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2149.448,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2160.745,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2370.02,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2125.471,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2125.105,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1795.448,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1792.173,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1795.046,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1795.921,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1793.492,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1794.268,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1795.235,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1792.405,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1657.138,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1656.563,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3314.001,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1656.667,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 2138.701,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1655.678,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3312.257,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1653.093,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1656.024,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3297.182,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3303.429,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1657.178,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1670.781,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1655.864,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1655.356,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 3328.646,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1794.456,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1796.91,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1795.289,
+            "min": 1500.0,
+            "max": 2250.0
+        },
+        {
+            "current": 1793.508,
+            "min": 1500.0,
+            "max": 2250.0
+        }
+    ],
+    "disk": {
+        "/": {
+            "total": 1757.8785285949707,
+            "used": 1663.5053787231445
+        }
+    },
+    "gpu": "NVIDIA A100-SXM4-80GB",
+    "gpu_count": 5,
+    "gpu_devices": [
+        {
+            "name": "NVIDIA A100-SXM4-80GB",
+            "memory_total": 85899345920
+        },
+        {
+            "name": "NVIDIA A100-SXM4-80GB",
+            "memory_total": 85899345920
+        },
+        {
+            "name": "NVIDIA A100-SXM4-80GB",
+            "memory_total": 85899345920
+        },
+        {
+            "name": "NVIDIA DGX Display",
+            "memory_total": 4294967296
+        },
+        {
+            "name": "NVIDIA A100-SXM4-80GB",
+            "memory_total": 85899345920
+        }
+    ],
+    "memory": {
+        "total": 503.5396919250488
+    }
+}

wandb/run-20240513_205248-d781ddha/files/wandb-summary.json

@@ -0,0 +1 @@
+{"_wandb": {"runtime": 215}}

wandb/run-20240513_205248-d781ddha/logs/debug-internal.log

@@ -0,0 +1,554 @@
+2024-05-13 20:52:48,872 INFO    StreamThr :1257957 [internal.py:wandb_internal():85] W&B internal server running at pid: 1257957, started at: 2024-05-13 20:52:48.871624
+2024-05-13 20:52:48,874 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status
+2024-05-13 20:52:48,876 INFO    WriterThread:1257957 [datastore.py:open_for_write():87] open: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/run-d781ddha.wandb
+2024-05-13 20:52:48,877 DEBUG   SenderThread:1257957 [sender.py:send():378] send: header
+2024-05-13 20:52:48,877 DEBUG   SenderThread:1257957 [sender.py:send():378] send: run
+2024-05-13 20:52:49,236 INFO    SenderThread:1257957 [dir_watcher.py:__init__():211] watching files in: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files
+2024-05-13 20:52:49,236 INFO    SenderThread:1257957 [sender.py:_start_run_threads():1123] run started: d781ddha with start time 1715626368.870296
+2024-05-13 20:52:49,241 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: check_version
+2024-05-13 20:52:49,241 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: check_version
+2024-05-13 20:52:49,306 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: run_start
+2024-05-13 20:52:49,337 DEBUG   HandlerThread:1257957 [system_info.py:__init__():26] System info init
+2024-05-13 20:52:49,337 DEBUG   HandlerThread:1257957 [system_info.py:__init__():41] System info init done
+2024-05-13 20:52:49,337 INFO    HandlerThread:1257957 [system_monitor.py:start():194] Starting system monitor
+2024-05-13 20:52:49,338 INFO    SystemMonitor:1257957 [system_monitor.py:_start():158] Starting system asset monitoring threads
+2024-05-13 20:52:49,338 INFO    HandlerThread:1257957 [system_monitor.py:probe():214] Collecting system info
+2024-05-13 20:52:49,338 INFO    SystemMonitor:1257957 [interfaces.py:start():188] Started cpu monitoring
+2024-05-13 20:52:49,339 INFO    SystemMonitor:1257957 [interfaces.py:start():188] Started disk monitoring
+2024-05-13 20:52:49,339 INFO    SystemMonitor:1257957 [interfaces.py:start():188] Started gpu monitoring
+2024-05-13 20:52:49,340 INFO    SystemMonitor:1257957 [interfaces.py:start():188] Started memory monitoring
+2024-05-13 20:52:49,341 INFO    SystemMonitor:1257957 [interfaces.py:start():188] Started network monitoring
+2024-05-13 20:52:49,375 DEBUG   HandlerThread:1257957 [system_info.py:probe():150] Probing system
+2024-05-13 20:52:49,377 DEBUG   HandlerThread:1257957 [system_info.py:_probe_git():135] Probing git
+2024-05-13 20:52:49,382 DEBUG   HandlerThread:1257957 [system_info.py:_probe_git():143] Probing git done
+2024-05-13 20:52:49,382 DEBUG   HandlerThread:1257957 [system_info.py:probe():198] Probing system done
+2024-05-13 20:52:49,382 DEBUG   HandlerThread:1257957 [system_monitor.py:probe():223] {'os': 'Linux-5.4.0-166-generic-x86_64-with-glibc2.31', 'python': '3.11.9', 'heartbeatAt': '2024-05-13T18:52:49.375376', 'startedAt': '2024-05-13T18:52:48.864820', 'docker': None, 'cuda': None, 'args': ('--model_name_or_path', 'parler-tts/parler_tts_mini_v0.1', '--feature_extractor_name', 'parler-tts/dac_44khZ_8kbps', '--description_tokenizer_name', 'parler-tts/parler_tts_mini_v0.1', '--prompt_tokenizer_name', 'parler-tts/parler_tts_mini_v0.1', '--report_to', 'wandb', '--overwrite_output_dir', 'true', '--train_dataset_name', 'sanchit-gandhi/expresso-concatenated-half-normal', '--train_metadata_dataset_name', 'sanchit-gandhi/expresso-concatenated-half-normal-tags-mistral', '--train_dataset_config_name', 'read', '--train_split_name', 'train', '--eval_dataset_name', 'sanchit-gandhi/expresso-concatenated-half-normal', '--eval_metadata_dataset_name', 'sanchit-gandhi/expresso-concatenated-half-normal-tags-mistral', '--eval_dataset_config_name', 'read', '--eval_split_name', 'train', '--max_eval_samples', '8', '--per_device_eval_batch_size', '16', '--target_audio_column_name', 'audio', '--description_column_name', 'text_description', '--prompt_column_name', 'text', '--max_duration_in_seconds', '30.0', '--min_duration_in_seconds', '2.0', '--max_text_length', '400', '--preprocessing_num_workers', '2', '--do_train', 'true', '--num_train_epochs', '8', '--gradient_accumulation_steps', '8', '--gradient_checkpointing', 'true', '--per_device_train_batch_size', '16', '--learning_rate', '0.00008', '--adam_beta1', '0.9', '--adam_beta2', '0.99', '--weight_decay', '0.01', '--lr_scheduler_type', 'cosine', '--warmup_steps', '250', '--logging_steps', '2', '--freeze_text_encoder', 'true', '--audio_encoder_per_device_batch_size', '4', '--dtype', 'bfloat16', '--seed', '456', '--output_dir', '../output_dir_training_concat/', '--temporary_save_to_disk', '../audio_code_tmp_concat/', '--save_to_disk', '../tmp_dataset_audio_concat/', '--dataloader_num_workers', '4', '--do_eval', '--predict_with_generate', '--include_inputs_for_metrics', '--save_strategy', 'steps', '--save_steps', '72', '--evaluation_strategy', 'epoch', '--save_total_limit', '5', '--group_by_length', 'true'), 'state': 'running', 'program': '/raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/./run_parler_tts_training.py', 'codePathLocal': 'run_parler_tts_training.py', 'codePath': 'run_parler_tts_training.py', 'git': {'remote': 'https://huggingface.co/sanchit-gandhi/parler-tts-mini-v0.1-expresso-concatenated', 'commit': '99f75adb5e13ee0ad87cce8deb3e71adc370b10e'}, 'email': 'sanchit@huggingface.co', 'root': '/raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated', 'host': 'hf-dgx-01', 'username': 'sanchit', 'executable': '/home/sanchit/miniconda3/envs/venv/bin/python', 'cpu_count': 64, 'cpu_count_logical': 128, 'cpu_freq': {'current': 2187.8050625000005, 'min': 1500.0, 'max': 2250.0}, 'cpu_freq_per_core': [{'current': 1713.071, 'min': 1500.0, 'max': 2250.0}, {'current': 1712.469, 'min': 1500.0, 'max': 2250.0}, {'current': 2280.258, 'min': 1500.0, 'max': 2250.0}, {'current': 1717.303, 'min': 1500.0, 'max': 2250.0}, {'current': 3312.578, 'min': 1500.0, 'max': 2250.0}, {'current': 3051.416, 'min': 1500.0, 'max': 2250.0}, {'current': 1658.955, 'min': 1500.0, 'max': 2250.0}, {'current': 1663.383, 'min': 1500.0, 'max': 2250.0}, {'current': 3332.084, 'min': 1500.0, 'max': 2250.0}, {'current': 1663.447, 'min': 1500.0, 'max': 2250.0}, {'current': 1663.494, 'min': 1500.0, 'max': 2250.0}, {'current': 1665.248, 'min': 1500.0, 'max': 2250.0}, {'current': 1793.968, 'min': 1500.0, 'max': 2250.0}, {'current': 1794.861, 'min': 1500.0, 'max': 2250.0}, {'current': 1793.472, 'min': 1500.0, 'max': 2250.0}, {'current': 1792.28, 'min': 1500.0, 'max': 2250.0}, {'current': 3341.098, 'min': 1500.0, 'max': 2250.0}, {'current': 1659.571, 'min': 1500.0, 'max': 2250.0}, {'current': 2090.509, 'min': 1500.0, 'max': 2250.0}, {'current': 3330.01, 'min': 1500.0, 'max': 2250.0}, {'current': 1661.972, 'min': 1500.0, 'max': 2250.0}, {'current': 1661.373, 'min': 1500.0, 'max': 2250.0}, {'current': 2744.439, 'min': 1500.0, 'max': 2250.0}, {'current': 3341.302, 'min': 1500.0, 'max': 2250.0}, {'current': 1931.367, 'min': 1500.0, 'max': 2250.0}, {'current': 1663.659, 'min': 1500.0, 'max': 2250.0}, {'current': 3325.735, 'min': 1500.0, 'max': 2250.0}, {'current': 1663.731, 'min': 1500.0, 'max': 2250.0}, {'current': 1715.888, 'min': 1500.0, 'max': 2250.0}, {'current': 1713.973, 'min': 1500.0, 'max': 2250.0}, {'current': 2361.772, 'min': 1500.0, 'max': 2250.0}, {'current': 1717.358, 'min': 1500.0, 'max': 2250.0}, {'current': 1716.521, 'min': 1500.0, 'max': 2250.0}, {'current': 2341.51, 'min': 1500.0, 'max': 2250.0}, {'current': 1715.513, 'min': 1500.0, 'max': 2250.0}, {'current': 1711.953, 'min': 1500.0, 'max': 2250.0}, {'current': 1793.443, 'min': 1500.0, 'max': 2250.0}, {'current': 1796.101, 'min': 1500.0, 'max': 2250.0}, {'current': 1794.087, 'min': 1500.0, 'max': 2250.0}, {'current': 1794.423, 'min': 1500.0, 'max': 2250.0}, {'current': 1790.765, 'min': 1500.0, 'max': 2250.0}, {'current': 1796.312, 'min': 1500.0, 'max': 2250.0}, {'current': 1792.734, 'min': 1500.0, 'max': 2250.0}, {'current': 1794.642, 'min': 1500.0, 'max': 2250.0}, {'current': 1659.721, 'min': 1500.0, 'max': 2250.0}, {'current': 1663.181, 'min': 1500.0, 'max': 2250.0}, {'current': 3315.724, 'min': 1500.0, 'max': 2250.0}, {'current': 1660.427, 'min': 1500.0, 'max': 2250.0}, {'current': 2966.59, 'min': 1500.0, 'max': 2250.0}, {'current': 1668.725, 'min': 1500.0, 'max': 2250.0}, {'current': 3300.161, 'min': 1500.0, 'max': 2250.0}, {'current': 1671.705, 'min': 1500.0, 'max': 2250.0}, {'current': 1661.312, 'min': 1500.0, 'max': 2250.0}, {'current': 3339.377, 'min': 1500.0, 'max': 2250.0}, {'current': 3333.308, 'min': 1500.0, 'max': 2250.0}, {'current': 1662.747, 'min': 1500.0, 'max': 2250.0}, {'current': 1660.781, 'min': 1500.0, 'max': 2250.0}, {'current': 1660.176, 'min': 1500.0, 'max': 2250.0}, {'current': 1664.39, 'min': 1500.0, 'max': 2250.0}, {'current': 3336.046, 'min': 1500.0, 'max': 2250.0}, {'current': 1793.809, 'min': 1500.0, 'max': 2250.0}, {'current': 1794.079, 'min': 1500.0, 'max': 2250.0}, {'current': 1594.389, 'min': 1500.0, 'max': 2250.0}, {'current': 1622.233, 'min': 1500.0, 'max': 2250.0}, {'current': 2414.464, 'min': 1500.0, 'max': 2250.0}, {'current': 2429.082, 'min': 1500.0, 'max': 2250.0}, {'current': 2278.496, 'min': 1500.0, 'max': 2250.0}, {'current': 2421.333, 'min': 1500.0, 'max': 2250.0}, {'current': 3330.367, 'min': 1500.0, 'max': 2250.0}, {'current': 2412.212, 'min': 1500.0, 'max': 2250.0}, {'current': 3102.925, 'min': 1500.0, 'max': 2250.0}, {'current': 2356.416, 'min': 1500.0, 'max': 2250.0}, {'current': 3312.565, 'min': 1500.0, 'max': 2250.0}, {'current': 3044.183, 'min': 1500.0, 'max': 2250.0}, {'current': 3296.93, 'min': 1500.0, 'max': 2250.0}, {'current': 3060.353, 'min': 1500.0, 'max': 2250.0}, {'current': 2411.85, 'min': 1500.0, 'max': 2250.0}, {'current': 2433.228, 'min': 1500.0, 'max': 2250.0}, {'current': 2464.094, 'min': 1500.0, 'max': 2250.0}, {'current': 2438.304, 'min': 1500.0, 'max': 2250.0}, {'current': 3306.628, 'min': 1500.0, 'max': 2250.0}, {'current': 3318.093, 'min': 1500.0, 'max': 2250.0}, {'current': 2748.09, 'min': 1500.0, 'max': 2250.0}, {'current': 3318.082, 'min': 1500.0, 'max': 2250.0}, {'current': 3316.036, 'min': 1500.0, 'max': 2250.0}, {'current': 3318.945, 'min': 1500.0, 'max': 2250.0}, {'current': 3313.086, 'min': 1500.0, 'max': 2250.0}, {'current': 3340.023, 'min': 1500.0, 'max': 2250.0}, {'current': 3306.8, 'min': 1500.0, 'max': 2250.0}, {'current': 2153.453, 'min': 1500.0, 'max': 2250.0}, {'current': 2690.778, 'min': 1500.0, 'max': 2250.0}, {'current': 2155.725, 'min': 1500.0, 'max': 2250.0}, {'current': 2149.357, 'min': 1500.0, 'max': 2250.0}, {'current': 2147.249, 'min': 1500.0, 'max': 2250.0}, {'current': 2359.311, 'min': 1500.0, 'max': 2250.0}, {'current': 2149.448, 'min': 1500.0, 'max': 2250.0}, {'current': 2160.745, 'min': 1500.0, 'max': 2250.0}, {'current': 2370.02, 'min': 1500.0, 'max': 2250.0}, {'current': 2125.471, 'min': 1500.0, 'max': 2250.0}, {'current': 2125.105, 'min': 1500.0, 'max': 2250.0}, {'current': 1795.448, 'min': 1500.0, 'max': 2250.0}, {'current': 1792.173, 'min': 1500.0, 'max': 2250.0}, {'current': 1795.046, 'min': 1500.0, 'max': 2250.0}, {'current': 1795.921, 'min': 1500.0, 'max': 2250.0}, {'current': 1793.492, 'min': 1500.0, 'max': 2250.0}, {'current': 1794.268, 'min': 1500.0, 'max': 2250.0}, {'current': 1795.235, 'min': 1500.0, 'max': 2250.0}, {'current': 1792.405, 'min': 1500.0, 'max': 2250.0}, {'current': 1657.138, 'min': 1500.0, 'max': 2250.0}, {'current': 1656.563, 'min': 1500.0, 'max': 2250.0}, {'current': 3314.001, 'min': 1500.0, 'max': 2250.0}, {'current': 1656.667, 'min': 1500.0, 'max': 2250.0}, {'current': 2138.701, 'min': 1500.0, 'max': 2250.0}, {'current': 1655.678, 'min': 1500.0, 'max': 2250.0}, {'current': 3312.257, 'min': 1500.0, 'max': 2250.0}, {'current': 1653.093, 'min': 1500.0, 'max': 2250.0}, {'current': 1656.024, 'min': 1500.0, 'max': 2250.0}, {'current': 3297.182, 'min': 1500.0, 'max': 2250.0}, {'current': 3303.429, 'min': 1500.0, 'max': 2250.0}, {'current': 1657.178, 'min': 1500.0, 'max': 2250.0}, {'current': 1670.781, 'min': 1500.0, 'max': 2250.0}, {'current': 1655.864, 'min': 1500.0, 'max': 2250.0}, {'current': 1655.356, 'min': 1500.0, 'max': 2250.0}, {'current': 3328.646, 'min': 1500.0, 'max': 2250.0}, {'current': 1794.456, 'min': 1500.0, 'max': 2250.0}, {'current': 1796.91, 'min': 1500.0, 'max': 2250.0}, {'current': 1795.289, 'min': 1500.0, 'max': 2250.0}, {'current': 1793.508, 'min': 1500.0, 'max': 2250.0}], 'disk': {'/': {'total': 1757.8785285949707, 'used': 1663.5053787231445}}, 'gpu': 'NVIDIA A100-SXM4-80GB', 'gpu_count': 5, 'gpu_devices': [{'name': 'NVIDIA A100-SXM4-80GB', 'memory_total': 85899345920}, {'name': 'NVIDIA A100-SXM4-80GB', 'memory_total': 85899345920}, {'name': 'NVIDIA A100-SXM4-80GB', 'memory_total': 85899345920}, {'name': 'NVIDIA DGX Display', 'memory_total': 4294967296}, {'name': 'NVIDIA A100-SXM4-80GB', 'memory_total': 85899345920}], 'memory': {'total': 503.5396919250488}}
+2024-05-13 20:52:49,382 INFO    HandlerThread:1257957 [system_monitor.py:probe():224] Finished collecting system info
+2024-05-13 20:52:49,382 INFO    HandlerThread:1257957 [system_monitor.py:probe():227] Publishing system info
+2024-05-13 20:52:49,382 DEBUG   HandlerThread:1257957 [system_info.py:_save_conda():207] Saving list of conda packages installed into the current environment
+2024-05-13 20:52:49,396 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:52:49,406 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:52:50,237 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_created():271] file/dir created: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/conda-environment.yaml
+2024-05-13 20:52:51,425 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:52:51,442 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:52:51,753 DEBUG   HandlerThread:1257957 [system_info.py:_save_conda():222] Saving conda packages done
+2024-05-13 20:52:51,754 INFO    HandlerThread:1257957 [system_monitor.py:probe():229] Finished publishing system info
+2024-05-13 20:52:51,762 DEBUG   SenderThread:1257957 [sender.py:send():378] send: files
+2024-05-13 20:52:51,762 INFO    SenderThread:1257957 [sender.py:_save_file():1389] saving file wandb-metadata.json with policy now
+2024-05-13 20:52:51,875 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: python_packages
+2024-05-13 20:52:51,875 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:52:51,876 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: python_packages
+2024-05-13 20:52:51,879 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:52:52,070 DEBUG   SenderThread:1257957 [sender.py:send():378] send: telemetry
+2024-05-13 20:52:52,070 DEBUG   SenderThread:1257957 [sender.py:send():378] send: config
+2024-05-13 20:52:52,237 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/conda-environment.yaml
+2024-05-13 20:52:52,237 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_created():271] file/dir created: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/requirements.txt
+2024-05-13 20:52:52,237 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_created():271] file/dir created: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/wandb-metadata.json
+2024-05-13 20:52:52,237 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_created():271] file/dir created: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:52:52,289 INFO    wandb-upload_0:1257957 [upload_job.py:push():130] Uploaded file /tmp/tmpapx4bmqmwandb/jir2ww2a-wandb-metadata.json
+2024-05-13 20:52:54,154 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:52:54,238 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:52:54,341 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:52:54,362 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:52:56,238 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:52:56,402 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:52:56,412 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:52:58,239 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:52:59,391 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:52:59,413 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:52:59,600 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:53:00,240 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:01,439 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:01,466 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:02,240 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:04,240 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:04,527 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:04,551 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:04,728 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:53:06,241 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:06,618 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:06,629 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:06,875 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:53:06,876 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:53:09,448 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:09,475 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:09,745 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:53:10,242 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:11,529 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:11,551 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:12,242 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:14,349 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:14,382 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:15,697 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:53:16,244 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:16,452 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:16,465 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:18,244 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:19,399 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:19,429 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:20,953 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:53:21,245 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/config.yaml
+2024-05-13 20:53:21,469 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:21,488 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:21,875 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:53:21,876 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:53:22,246 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:24,246 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:24,383 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:24,407 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:26,755 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:53:28,202 ERROR   gpu       :1257957 [interfaces.py:aggregate():159] Failed to serialize metric: division by zero
+2024-05-13 20:53:28,227 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:28,242 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:30,248 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:31,218 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:31,240 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:31,787 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:53:32,249 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:33,272 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:33,297 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:36,218 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:36,250 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:36,253 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:36,876 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:53:36,876 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:53:37,092 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:53:38,250 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:38,303 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:38,313 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:40,251 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:41,370 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:41,397 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:42,166 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:53:42,252 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:43,423 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:43,441 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:44,252 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:46,253 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:46,761 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:46,792 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:47,210 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:53:48,254 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:48,869 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:48,922 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:49,341 DEBUG   SystemMonitor:1257957 [system_monitor.py:_start():172] Starting system metrics aggregation loop
+2024-05-13 20:53:49,346 DEBUG   SenderThread:1257957 [sender.py:send():378] send: stats
+2024-05-13 20:53:50,254 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:50,950 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:50,971 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:51,875 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:53:51,876 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:53:52,255 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:52,329 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:53:53,011 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:53,045 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:54,255 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:56,224 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:56,235 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:56,256 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:57,574 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:53:58,257 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:53:58,298 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:53:58,323 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:00,257 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:00,372 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:00,381 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:02,257 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:02,473 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:02,486 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:02,652 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:54:04,260 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:06,260 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:06,527 ERROR   gpu       :1257957 [interfaces.py:aggregate():159] Failed to serialize metric: division by zero
+2024-05-13 20:54:06,875 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:54:06,875 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:54:07,729 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:54:07,954 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:07,968 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:08,261 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:09,996 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:10,009 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:10,262 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:12,063 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:12,083 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:12,262 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:12,769 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:54:14,119 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:14,133 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:14,263 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:16,172 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:16,196 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:16,263 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:17,810 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:54:18,232 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:18,250 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:18,264 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:19,348 DEBUG   SenderThread:1257957 [sender.py:send():378] send: stats
+2024-05-13 20:54:20,264 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:20,272 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:20,282 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:21,875 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:54:21,876 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:54:22,266 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:22,303 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:22,313 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:22,858 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:54:24,266 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:24,334 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:24,349 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:26,267 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:26,379 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:26,401 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:27,908 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:54:28,267 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:28,422 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:28,440 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:30,268 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:30,472 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:30,484 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:32,269 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:32,538 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:32,552 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:33,018 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:54:34,269 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:34,686 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:34,726 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:36,269 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:36,860 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:36,875 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:54:36,876 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:54:36,880 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:38,057 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:54:38,270 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:40,270 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:40,368 ERROR   gpu       :1257957 [interfaces.py:aggregate():159] Failed to serialize metric: division by zero
+2024-05-13 20:54:41,242 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:41,259 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:42,271 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:43,097 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:54:43,484 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:43,506 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:44,272 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:45,567 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:45,583 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:46,272 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:47,618 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:47,632 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:48,273 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:48,308 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:54:49,350 DEBUG   SenderThread:1257957 [sender.py:send():378] send: stats
+2024-05-13 20:54:49,662 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:49,677 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:50,273 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:51,727 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:51,748 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:51,876 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:54:51,876 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:54:52,274 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:53,618 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:54:53,953 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:53,975 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:54,275 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:56,117 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:56,151 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:56,275 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:58,276 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:54:58,370 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:58,407 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:54:58,644 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:55:00,276 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:00,559 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:00,601 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:02,277 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:02,776 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:02,807 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:03,686 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:55:04,277 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:04,969 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:05,009 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:06,278 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:06,875 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:55:06,875 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:55:07,138 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:07,162 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:08,278 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:08,783 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:55:09,293 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:09,336 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:10,279 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:11,446 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:11,486 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:12,280 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:13,807 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:55:14,280 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:15,679 ERROR   gpu       :1257957 [interfaces.py:aggregate():159] Failed to serialize metric: division by zero
+2024-05-13 20:55:15,736 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:15,758 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:16,281 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:17,815 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:17,846 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:18,281 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:18,856 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:55:19,353 DEBUG   SenderThread:1257957 [sender.py:send():378] send: stats
+2024-05-13 20:55:19,930 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:19,956 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:20,282 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:21,876 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:55:21,876 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:55:22,142 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:22,170 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:22,282 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:23,933 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:55:24,253 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:24,274 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:24,283 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:26,283 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:26,332 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:26,343 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:28,285 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:28,419 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:28,451 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:29,032 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:55:30,285 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:30,624 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:30,647 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:32,286 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:32,797 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:32,841 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:34,229 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:55:34,286 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:35,011 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:35,035 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:36,287 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:36,876 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:55:36,876 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:55:37,215 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:37,236 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:38,287 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:39,254 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:55:39,435 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:39,455 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:40,288 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:41,535 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:41,547 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:42,289 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:43,627 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:43,638 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:44,279 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:55:44,289 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:45,711 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:45,770 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:46,290 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:48,290 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:49,355 DEBUG   SenderThread:1257957 [sender.py:send():378] send: stats
+2024-05-13 20:55:49,356 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:55:50,017 ERROR   gpu       :1257957 [interfaces.py:aggregate():159] Failed to serialize metric: division by zero
+2024-05-13 20:55:50,063 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:50,091 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:50,291 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:51,876 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:55:51,876 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:55:52,211 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:52,242 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:52,291 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:54,292 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:54,379 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:54,387 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:55:54,419 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:56,292 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:56,566 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:56,604 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:58,293 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:55:58,654 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:58,674 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:55:59,594 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:56:00,294 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:56:00,708 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:00,723 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:02,294 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:56:02,766 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:02,781 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:04,294 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:56:04,633 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:56:04,818 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:04,829 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:06,295 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:56:06,853 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:06,862 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:06,876 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:56:06,876 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:56:08,295 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:56:08,898 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:08,916 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:09,783 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:56:10,296 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:56:10,950 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:11,697 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:12,297 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:56:14,054 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:14,064 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:14,297 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:56:14,910 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:56:16,101 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:16,114 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:16,298 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:56:18,143 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:18,154 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:18,298 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:56:19,357 DEBUG   SenderThread:1257957 [sender.py:send():378] send: stats
+2024-05-13 20:56:19,989 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:56:20,194 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:20,212 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:20,299 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:56:21,876 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: stop_status
+2024-05-13 20:56:21,876 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: stop_status
+2024-05-13 20:56:22,299 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:56:24,300 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:56:24,913 ERROR   gpu       :1257957 [interfaces.py:aggregate():159] Failed to serialize metric: division by zero
+2024-05-13 20:56:25,185 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:56:25,189 DEBUG   SenderThread:1257957 [sender.py:send():378] send: exit
+2024-05-13 20:56:25,189 INFO    SenderThread:1257957 [sender.py:send_exit():585] handling exit code: 1
+2024-05-13 20:56:25,189 INFO    SenderThread:1257957 [sender.py:send_exit():587] handling runtime: 215
+2024-05-13 20:56:25,189 INFO    SenderThread:1257957 [sender.py:_save_file():1389] saving file wandb-summary.json with policy end
+2024-05-13 20:56:25,189 INFO    SenderThread:1257957 [sender.py:send_exit():593] send defer
+2024-05-13 20:56:25,189 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:56:25,189 INFO    HandlerThread:1257957 [handler.py:handle_request_defer():184] handle defer: 0
+2024-05-13 20:56:25,190 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:56:25,190 INFO    SenderThread:1257957 [sender.py:send_request_defer():609] handle sender defer: 0
+2024-05-13 20:56:25,190 INFO    SenderThread:1257957 [sender.py:transition_state():613] send defer: 1
+2024-05-13 20:56:25,190 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:56:25,190 INFO    HandlerThread:1257957 [handler.py:handle_request_defer():184] handle defer: 1
+2024-05-13 20:56:25,190 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:56:25,190 INFO    SenderThread:1257957 [sender.py:send_request_defer():609] handle sender defer: 1
+2024-05-13 20:56:25,190 INFO    SenderThread:1257957 [sender.py:transition_state():613] send defer: 2
+2024-05-13 20:56:25,190 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:56:25,190 INFO    HandlerThread:1257957 [handler.py:handle_request_defer():184] handle defer: 2
+2024-05-13 20:56:25,190 INFO    HandlerThread:1257957 [system_monitor.py:finish():203] Stopping system monitor
+2024-05-13 20:56:25,190 DEBUG   SystemMonitor:1257957 [system_monitor.py:_start():179] Finished system metrics aggregation loop
+2024-05-13 20:56:25,190 DEBUG   SystemMonitor:1257957 [system_monitor.py:_start():183] Publishing last batch of metrics
+2024-05-13 20:56:25,193 INFO    HandlerThread:1257957 [interfaces.py:finish():200] Joined cpu monitor
+2024-05-13 20:56:25,193 INFO    HandlerThread:1257957 [interfaces.py:finish():200] Joined disk monitor
+2024-05-13 20:56:25,231 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:25,252 ERROR   gpu       :1257957 [interfaces.py:monitor():142] Failed to sample metric: Not Supported
+2024-05-13 20:56:25,300 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_created():271] file/dir created: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/wandb-summary.json
+2024-05-13 20:56:26,300 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:56:26,956 ERROR   gpu       :1257957 [interfaces.py:aggregate():159] Failed to serialize metric: division by zero
+2024-05-13 20:56:26,956 INFO    HandlerThread:1257957 [interfaces.py:finish():200] Joined gpu monitor
+2024-05-13 20:56:26,956 INFO    HandlerThread:1257957 [interfaces.py:finish():200] Joined memory monitor
+2024-05-13 20:56:26,956 INFO    HandlerThread:1257957 [interfaces.py:finish():200] Joined network monitor
+2024-05-13 20:56:26,956 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: poll_exit
+2024-05-13 20:56:26,958 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:56:26,958 INFO    SenderThread:1257957 [sender.py:send_request_defer():609] handle sender defer: 2
+2024-05-13 20:56:26,958 INFO    SenderThread:1257957 [sender.py:transition_state():613] send defer: 3
+2024-05-13 20:56:26,958 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:56:26,958 INFO    HandlerThread:1257957 [handler.py:handle_request_defer():184] handle defer: 3
+2024-05-13 20:56:26,959 DEBUG   SenderThread:1257957 [sender.py:send():378] send: stats
+2024-05-13 20:56:26,959 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: poll_exit
+2024-05-13 20:56:26,960 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:56:26,960 INFO    SenderThread:1257957 [sender.py:send_request_defer():609] handle sender defer: 3
+2024-05-13 20:56:26,960 INFO    SenderThread:1257957 [sender.py:transition_state():613] send defer: 4
+2024-05-13 20:56:26,960 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:56:26,960 INFO    HandlerThread:1257957 [handler.py:handle_request_defer():184] handle defer: 4
+2024-05-13 20:56:26,960 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:56:26,960 INFO    SenderThread:1257957 [sender.py:send_request_defer():609] handle sender defer: 4
+2024-05-13 20:56:26,960 INFO    SenderThread:1257957 [sender.py:transition_state():613] send defer: 5
+2024-05-13 20:56:26,960 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:56:26,960 INFO    HandlerThread:1257957 [handler.py:handle_request_defer():184] handle defer: 5
+2024-05-13 20:56:26,961 DEBUG   SenderThread:1257957 [sender.py:send():378] send: summary
+2024-05-13 20:56:26,961 INFO    SenderThread:1257957 [sender.py:_save_file():1389] saving file wandb-summary.json with policy end
+2024-05-13 20:56:26,961 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:56:26,961 INFO    SenderThread:1257957 [sender.py:send_request_defer():609] handle sender defer: 5
+2024-05-13 20:56:26,961 INFO    SenderThread:1257957 [sender.py:transition_state():613] send defer: 6
+2024-05-13 20:56:26,961 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:56:26,961 INFO    HandlerThread:1257957 [handler.py:handle_request_defer():184] handle defer: 6
+2024-05-13 20:56:26,961 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:56:26,961 INFO    SenderThread:1257957 [sender.py:send_request_defer():609] handle sender defer: 6
+2024-05-13 20:56:26,961 INFO    SenderThread:1257957 [sender.py:transition_state():613] send defer: 7
+2024-05-13 20:56:26,962 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: status_report
+2024-05-13 20:56:26,962 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:56:26,962 INFO    HandlerThread:1257957 [handler.py:handle_request_defer():184] handle defer: 7
+2024-05-13 20:56:26,962 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:56:26,962 INFO    SenderThread:1257957 [sender.py:send_request_defer():609] handle sender defer: 7
+2024-05-13 20:56:27,189 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: poll_exit
+2024-05-13 20:56:27,301 INFO    Thread-12 :1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/wandb-summary.json
+2024-05-13 20:56:28,505 INFO    SenderThread:1257957 [sender.py:transition_state():613] send defer: 8
+2024-05-13 20:56:28,506 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: poll_exit
+2024-05-13 20:56:28,506 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:56:28,506 INFO    HandlerThread:1257957 [handler.py:handle_request_defer():184] handle defer: 8
+2024-05-13 20:56:28,506 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:56:28,506 INFO    SenderThread:1257957 [sender.py:send_request_defer():609] handle sender defer: 8
+2024-05-13 20:56:28,506 INFO    SenderThread:1257957 [job_builder.py:build():432] Attempting to build job artifact
+2024-05-13 20:56:28,507 INFO    SenderThread:1257957 [job_builder.py:_get_source_type():565] is repo sourced job
+2024-05-13 20:56:28,530 INFO    SenderThread:1257957 [job_builder.py:build():541] adding wandb-job metadata file
+2024-05-13 20:56:28,532 INFO    SenderThread:1257957 [sender.py:transition_state():613] send defer: 9
+2024-05-13 20:56:28,532 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:56:28,532 INFO    HandlerThread:1257957 [handler.py:handle_request_defer():184] handle defer: 9
+2024-05-13 20:56:28,533 DEBUG   SenderThread:1257957 [sender.py:send():378] send: artifact
+2024-05-13 20:56:29,160 INFO    SenderThread:1257957 [sender.py:send_artifact():1467] sent artifact job-https___huggingface.co_sanchit-gandhi_parler-tts-mini-v0.1-expresso-concatenated_run_parler_tts_training.py - {'id': 'QXJ0aWZhY3Q6ODM0NzI5Njgw', 'state': 'COMMITTED', 'artifactSequence': {'id': 'QXJ0aWZhY3RDb2xsZWN0aW9uOjE3NDIzMTI1Mw==', 'latestArtifact': {'id': 'QXJ0aWZhY3Q6ODM0NzI5Njgw', 'versionIndex': 0}}}
+2024-05-13 20:56:29,160 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:56:29,160 INFO    SenderThread:1257957 [sender.py:send_request_defer():609] handle sender defer: 9
+2024-05-13 20:56:29,160 INFO    SenderThread:1257957 [dir_watcher.py:finish():358] shutting down directory watcher
+2024-05-13 20:56:29,191 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: poll_exit
+2024-05-13 20:56:29,302 INFO    SenderThread:1257957 [dir_watcher.py:_on_file_modified():288] file/dir modified: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:56:29,303 INFO    SenderThread:1257957 [dir_watcher.py:finish():388] scan: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files
+2024-05-13 20:56:29,303 INFO    SenderThread:1257957 [dir_watcher.py:finish():402] scan save: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/conda-environment.yaml conda-environment.yaml
+2024-05-13 20:56:29,303 INFO    SenderThread:1257957 [dir_watcher.py:finish():402] scan save: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/wandb-summary.json wandb-summary.json
+2024-05-13 20:56:29,303 INFO    SenderThread:1257957 [dir_watcher.py:finish():402] scan save: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log output.log
+2024-05-13 20:56:29,303 INFO    SenderThread:1257957 [dir_watcher.py:finish():402] scan save: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/requirements.txt requirements.txt
+2024-05-13 20:56:29,303 INFO    SenderThread:1257957 [dir_watcher.py:finish():402] scan save: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/config.yaml config.yaml
+2024-05-13 20:56:29,303 INFO    SenderThread:1257957 [dir_watcher.py:finish():402] scan save: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/wandb-metadata.json wandb-metadata.json
+2024-05-13 20:56:29,303 INFO    SenderThread:1257957 [sender.py:transition_state():613] send defer: 10
+2024-05-13 20:56:29,303 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: poll_exit
+2024-05-13 20:56:29,304 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:56:29,304 INFO    HandlerThread:1257957 [handler.py:handle_request_defer():184] handle defer: 10
+2024-05-13 20:56:29,310 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:56:29,310 INFO    SenderThread:1257957 [sender.py:send_request_defer():609] handle sender defer: 10
+2024-05-13 20:56:29,310 INFO    SenderThread:1257957 [file_pusher.py:finish():169] shutting down file pusher
+2024-05-13 20:56:29,628 INFO    wandb-upload_1:1257957 [upload_job.py:push():130] Uploaded file /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/wandb-summary.json
+2024-05-13 20:56:29,688 INFO    wandb-upload_0:1257957 [upload_job.py:push():130] Uploaded file /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/conda-environment.yaml
+2024-05-13 20:56:29,743 INFO    wandb-upload_4:1257957 [upload_job.py:push():130] Uploaded file /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/config.yaml
+2024-05-13 20:56:29,770 INFO    wandb-upload_3:1257957 [upload_job.py:push():130] Uploaded file /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/requirements.txt
+2024-05-13 20:56:29,817 INFO    wandb-upload_2:1257957 [upload_job.py:push():130] Uploaded file /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/files/output.log
+2024-05-13 20:56:30,017 INFO    Thread-11 (_thread_body):1257957 [sender.py:transition_state():613] send defer: 11
+2024-05-13 20:56:30,018 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:56:30,018 INFO    HandlerThread:1257957 [handler.py:handle_request_defer():184] handle defer: 11
+2024-05-13 20:56:30,018 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:56:30,018 INFO    SenderThread:1257957 [sender.py:send_request_defer():609] handle sender defer: 11
+2024-05-13 20:56:30,018 INFO    SenderThread:1257957 [file_pusher.py:join():175] waiting for file pusher
+2024-05-13 20:56:30,019 INFO    SenderThread:1257957 [sender.py:transition_state():613] send defer: 12
+2024-05-13 20:56:30,019 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:56:30,019 INFO    HandlerThread:1257957 [handler.py:handle_request_defer():184] handle defer: 12
+2024-05-13 20:56:30,019 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:56:30,019 INFO    SenderThread:1257957 [sender.py:send_request_defer():609] handle sender defer: 12
+2024-05-13 20:56:30,019 INFO    SenderThread:1257957 [file_stream.py:finish():601] file stream finish called
+2024-05-13 20:56:30,156 INFO    SenderThread:1257957 [file_stream.py:finish():605] file stream finish is done
+2024-05-13 20:56:30,156 INFO    SenderThread:1257957 [sender.py:transition_state():613] send defer: 13
+2024-05-13 20:56:30,156 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:56:30,156 INFO    HandlerThread:1257957 [handler.py:handle_request_defer():184] handle defer: 13
+2024-05-13 20:56:30,156 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:56:30,156 INFO    SenderThread:1257957 [sender.py:send_request_defer():609] handle sender defer: 13
+2024-05-13 20:56:30,157 INFO    SenderThread:1257957 [sender.py:transition_state():613] send defer: 14
+2024-05-13 20:56:30,157 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: defer
+2024-05-13 20:56:30,157 INFO    HandlerThread:1257957 [handler.py:handle_request_defer():184] handle defer: 14
+2024-05-13 20:56:30,157 DEBUG   SenderThread:1257957 [sender.py:send():378] send: final
+2024-05-13 20:56:30,157 DEBUG   SenderThread:1257957 [sender.py:send():378] send: footer
+2024-05-13 20:56:30,157 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: defer
+2024-05-13 20:56:30,157 INFO    SenderThread:1257957 [sender.py:send_request_defer():609] handle sender defer: 14
+2024-05-13 20:56:30,158 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: poll_exit
+2024-05-13 20:56:30,158 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: poll_exit
+2024-05-13 20:56:30,159 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: poll_exit
+2024-05-13 20:56:30,159 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: poll_exit
+2024-05-13 20:56:30,159 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: server_info
+2024-05-13 20:56:30,159 DEBUG   SenderThread:1257957 [sender.py:send_request():405] send_request: server_info
+2024-05-13 20:56:30,159 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: get_summary
+2024-05-13 20:56:30,161 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: sampled_history
+2024-05-13 20:56:30,161 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: internal_messages
+2024-05-13 20:56:30,298 INFO    MainThread:1257957 [wandb_run.py:_footer_history_summary_info():3994] rendering history
+2024-05-13 20:56:30,299 INFO    MainThread:1257957 [wandb_run.py:_footer_history_summary_info():4026] rendering summary
+2024-05-13 20:56:30,299 INFO    MainThread:1257957 [wandb_run.py:_footer_sync_info():3953] logging synced files
+2024-05-13 20:56:30,299 DEBUG   HandlerThread:1257957 [handler.py:handle_request():158] handle_request: shutdown
+2024-05-13 20:56:30,299 INFO    HandlerThread:1257957 [handler.py:finish():882] shutting down handler
+2024-05-13 20:56:31,159 INFO    WriterThread:1257957 [datastore.py:close():296] close: /raid/sanchit/parler-tts-mini-v0.1-expresso-concatenated/wandb/run-20240513_205248-d781ddha/run-d781ddha.wandb
+2024-05-13 20:56:31,298 INFO    SenderThread:1257957 [sender.py:finish():1545] shutting down sender
+2024-05-13 20:56:31,298 INFO    SenderThread:1257957 [file_pusher.py:finish():169] shutting down file pusher
+2024-05-13 20:56:31,299 INFO    SenderThread:1257957 [file_pusher.py:join():175] waiting for file pusher