/home/victor/anaconda3/envs/suno_env_fa2_fa3_pt29/lib/python3.10/site-packages/torch/library.py:356: UserWarning: Warning only once for all operators,  other operators may also be overridden.
  Overriding a previously registered kernel for the same operator and the same dispatch key
  operator: flash_attn::_flash_attn_backward(Tensor dout, Tensor q, Tensor k, Tensor v, Tensor out, Tensor softmax_lse, Tensor(a6!)? dq, Tensor(a7!)? dk, Tensor(a8!)? dv, float dropout_p, float softmax_scale, bool causal, SymInt window_size_left, SymInt window_size_right, float softcap, Tensor? alibi_slopes, bool deterministic, Tensor? rng_state=None) -> Tensor
    registered at /home/victor/anaconda3/envs/suno_env_fa2_fa3_pt29/lib/python3.10/site-packages/torch/_library/custom_ops.py:922
  dispatch key: ADInplaceOrView
  previous kernel: no debug info
       new kernel: registered at /home/victor/anaconda3/envs/suno_env_fa2_fa3_pt29/lib/python3.10/site-packages/torch/_library/custom_ops.py:922 (Triggered internally at /pytorch/aten/src/ATen/core/dispatch/OperatorEntry.cpp:208.)
  self.m.impl(
/home/victor/anaconda3/envs/suno_env_fa2_fa3_pt29/lib/python3.10/site-packages/webrtcvad.py:1: UserWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html. The pkg_resources package is slated for removal as early as 2025-11-30. Refrain from using this package or pin to Setuptools<81.
  import pkg_resources
Loading configuration from configurator.py
Overriding: data_dir = /app2/suno/data/auk_v0
Overriding: train_metas_filename = metas_v8_tr_mini.jsonl
Overriding: batch_store_size = 1
Overriding: n_layer = 2
Overriding: n_head = 32
Overriding: d_head = 128
Overriding: block_size = 3200
Overriding: batch_size = 1
Overriding: allow_skip = False
Overriding: num_batches_to_profile = 50
Overriding: num_workers_oracle = 6

Final Configuration:
  data_dir: /app2/suno/data/auk_v0
  train_metas_filename: metas_v8_tr_mini.jsonl
  block_size: 3200
  batch_size: 1
  batch_store_size: 1
  n_layer: 2
  n_head: 32
  d_head: 128

================================================================================
DATA LOADER PROFILING
================================================================================
Data directory: /app2/suno/data/auk_v0
Batch size: 1
Batch size tokens: 3200
Batch store size: 1
Pack: True
Oracle workers: 6
Prefetch factor: 4
Number of batches: 50
================================================================================

Creating oracle dataset...
Traceback (most recent call last):
  File "/home/victor/neon/sunoGPT/profile_dataloader.py", line 457, in <module>
    profile_dataloader()
  File "/home/victor/neon/sunoGPT/profile_dataloader.py", line 268, in profile_dataloader
    oracle_dataset = AudioLoaderDataset(
TypeError: AudioLoaderDataset.__init__() got an unexpected keyword argument 'stem_active_sections_weight'