Update sample.py

YoungSeng · Jul 4, 2023 · 85f4096 · 85f4096
1 parent f61171e
commit 85f4096
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/main/mydiffusion_zeggs/sample.py b/main/mydiffusion_zeggs/sample.py
@@ -376,6 +376,8 @@ def main(args, save_dir, model_path, audio_path=None, mfcc_path=None, audiowavlm
     sample_fn = diffusion.p_sample_loop     # predict x_start
 
     style = style2onehot[audiowavlm_path.split('/')[-1].split('_')[1]]
+    # style = [0, 0, 1, 0, 0, 0]
+    # style = style2onehot['Neutral']
     print(style)
 
     wavlm_model = wavlm_init(mydevice)