shenyunhang commited on
Commit
82f2cfa
·
1 Parent(s): 52e4f53
vita_audio/data/dataset_base.py CHANGED
@@ -7,7 +7,6 @@ import re
7
  import traceback
8
  import uuid
9
 
10
- import cv2
11
  import numpy as np
12
  import torch
13
  import yaml
 
7
  import traceback
8
  import uuid
9
 
 
10
  import numpy as np
11
  import torch
12
  import yaml
vita_audio/data/dataset_cosyvoice2.py CHANGED
@@ -11,7 +11,6 @@ import traceback
11
  from collections import defaultdict
12
  from typing import Dict, List, Optional, Sequence
13
 
14
- import cv2
15
  import numpy as np
16
  import torch
17
  import transformers
 
11
  from collections import defaultdict
12
  from typing import Dict, List, Optional, Sequence
13
 
 
14
  import numpy as np
15
  import torch
16
  import transformers
vita_audio/data/dataset_deepseek.py CHANGED
@@ -11,7 +11,6 @@ import traceback
11
  from collections import defaultdict
12
  from typing import Dict, List, Optional, Sequence
13
 
14
- import cv2
15
  import numpy as np
16
  import torch
17
  import transformers
 
11
  from collections import defaultdict
12
  from typing import Dict, List, Optional, Sequence
13
 
 
14
  import numpy as np
15
  import torch
16
  import transformers
vita_audio/data/dataset_hunyuan.py CHANGED
@@ -11,7 +11,6 @@ import traceback
11
  from collections import defaultdict
12
  from typing import Dict, List, Optional, Sequence
13
 
14
- import cv2
15
  import numpy as np
16
  import torch
17
  import transformers
 
11
  from collections import defaultdict
12
  from typing import Dict, List, Optional, Sequence
13
 
 
14
  import numpy as np
15
  import torch
16
  import transformers
vita_audio/data/dataset_llama3.py CHANGED
@@ -11,7 +11,6 @@ import traceback
11
  from collections import defaultdict
12
  from typing import Dict, List, Optional, Sequence
13
 
14
- import cv2
15
  import numpy as np
16
  import torch
17
  import transformers
 
11
  from collections import defaultdict
12
  from typing import Dict, List, Optional, Sequence
13
 
 
14
  import numpy as np
15
  import torch
16
  import transformers
vita_audio/data/dataset_mistral.py CHANGED
@@ -9,7 +9,6 @@ import traceback
9
  from collections import defaultdict
10
  from typing import Dict, List, Optional, Sequence
11
 
12
- import cv2
13
  import numpy as np
14
  import torch
15
  import transformers
 
9
  from collections import defaultdict
10
  from typing import Dict, List, Optional, Sequence
11
 
 
12
  import numpy as np
13
  import torch
14
  import transformers
vita_audio/data/dataset_qwen2.py CHANGED
@@ -11,7 +11,6 @@ import traceback
11
  from collections import defaultdict
12
  from typing import Dict, List, Optional, Sequence
13
 
14
- import cv2
15
  import numpy as np
16
  import torch
17
  import transformers
 
11
  from collections import defaultdict
12
  from typing import Dict, List, Optional, Sequence
13
 
 
14
  import numpy as np
15
  import torch
16
  import transformers
vita_audio/data/processor/audio_processor.py CHANGED
@@ -2,8 +2,6 @@ import json
2
  import math
3
  import os
4
 
5
- import cv2
6
- import numpy as np
7
  import torch
8
 
9
  import natsort
 
2
  import math
3
  import os
4
 
 
 
5
  import torch
6
 
7
  import natsort
vita_audio/data/processor/image_processor.py CHANGED
@@ -1,7 +1,6 @@
1
  import math
2
  import os
3
 
4
- import cv2
5
  import numpy as np
6
  import torch
7
  from PIL import Image
 
1
  import math
2
  import os
3
 
 
4
  import numpy as np
5
  import torch
6
  from PIL import Image