Commit
•
167a352
1
Parent(s):
ccbcff5
Update audioldm_train/modules/clap/training/data.py
Browse files
audioldm_train/modules/clap/training/data.py
CHANGED
|
@@ -6,7 +6,7 @@ import os
|
|
| 6 |
import random
|
| 7 |
import h5py
|
| 8 |
from dataclasses import dataclass
|
| 9 |
-
from audioldm_train.modules.clap.training.params import parse_args
|
| 10 |
import braceexpand
|
| 11 |
import numpy as np
|
| 12 |
import pandas as pd
|
|
@@ -25,11 +25,11 @@ import io
|
|
| 25 |
from pathlib import Path
|
| 26 |
import wget
|
| 27 |
|
| 28 |
-
from audioldm_train.modules.clap.open_clip.utils import (
|
| 29 |
get_tar_path_from_dataset_name,
|
| 30 |
dataset_split,
|
| 31 |
)
|
| 32 |
-
from audioldm_train.modules.clap.open_clip.utils import load_p, load_class_label
|
| 33 |
import tempfile
|
| 34 |
import copy
|
| 35 |
|
|
@@ -43,7 +43,7 @@ try:
|
|
| 43 |
except ImportError:
|
| 44 |
torchaudio = None
|
| 45 |
|
| 46 |
-
from audioldm_train.modules.clap.open_clip import tokenize
|
| 47 |
|
| 48 |
|
| 49 |
def tokenizer(text):
|
|
|
|
| 6 |
import random
|
| 7 |
import h5py
|
| 8 |
from dataclasses import dataclass
|
| 9 |
+
from qa_mdt.audioldm_train.modules.clap.training.params import parse_args
|
| 10 |
import braceexpand
|
| 11 |
import numpy as np
|
| 12 |
import pandas as pd
|
|
|
|
| 25 |
from pathlib import Path
|
| 26 |
import wget
|
| 27 |
|
| 28 |
+
from qa_mdt.audioldm_train.modules.clap.open_clip.utils import (
|
| 29 |
get_tar_path_from_dataset_name,
|
| 30 |
dataset_split,
|
| 31 |
)
|
| 32 |
+
from qa_mdt.audioldm_train.modules.clap.open_clip.utils import load_p, load_class_label
|
| 33 |
import tempfile
|
| 34 |
import copy
|
| 35 |
|
|
|
|
| 43 |
except ImportError:
|
| 44 |
torchaudio = None
|
| 45 |
|
| 46 |
+
from qa_mdt.audioldm_train.modules.clap.open_clip import tokenize
|
| 47 |
|
| 48 |
|
| 49 |
def tokenizer(text):
|