Преглед изворни кода

fix: remove references to nonexistent keys in vqgan dataset test (#1206)

The __main__ test block accessed batch["features"] and
batch["feature_lengths"], but VQGANCollator only returns "audios"
and "audio_lengths", causing a KeyError when running the file directly.

Made-with: Cursor
huang yutong пре 2 недеља
родитељ
комит
0a35e1c77b
1 измењених фајлова са 0 додато и 2 уклоњено
  1. 0 2
      fish_speech/datasets/vqgan.py

+ 0 - 2
fish_speech/datasets/vqgan.py

@@ -141,7 +141,5 @@ if __name__ == "__main__":
 
     for batch in dataloader:
         print(batch["audios"].shape)
-        print(batch["features"].shape)
         print(batch["audio_lengths"])
-        print(batch["feature_lengths"])
         break