Update model.py

model.py

@@ -1517,10 +1517,10 @@ def main():
         text_dims=512,
         text_idx=4,
         act="swish",
-        debug={},
+        debug={},  #{"encoder", "decoder", "residual", "rotary"}, debug prints for specific modules
         cross_attn=True,
         f0_rotary=True, 
-        features = ["spectrogram"],
+        features = ["spectrogram"], # ["spectrogram", "waveform", "pitch"] any combo and order matters
         )
 
     sanity_check = False
@@ -1532,7 +1532,7 @@ def main():
         "waveforms": False,
         "pitch": False,
         "downsamples": False,
-        "f0": True,
+        "f0": True, # this needs to be true along with f0_rotary to pass f0 to rotary - Pitch passess as a feature
         "hilbert": False,
         "hop_length": 128,
         "fmin": 150,