disable jit

cosyvoice_rodis/flow/flow.py

@@ -3,6 +3,8 @@
 import logging
 import random
 from typing import Dict, Optional
+import os
+os.environ["TORCH_JIT_DISABLE"] = "1"
 import torch
 import torch.nn as nn
 from torch.nn import functional as F
@@ -157,6 +159,9 @@ class MaskedDiffWithXvec(torch.nn.Module):
         token = self.input_embedding(torch.clamp(token, min=0).long()) * mask  
 
         # text encode
+
+        print("token:", token, token_len, flow_emotion_embedding)
+        print("token shape:", token.shape(), token_len.shape(), flow_emotion_embedding.shape())
         h, h_lengths = self.encoder(token, token_len, flow_emotion_embedding)
         h = self.encoder_proj(h) #torch.Size([1, 358, 80])
         mel_len1, mel_len2 = prompt_feat.shape[1], int(token_len2 / self.input_frame_rate * 22050 / 256)