Commit
•
def46d7
1
Parent(s):
dc083ab
Update infer/infer_mos5.py
Browse files- infer/infer_mos5.py +3 -3
infer/infer_mos5.py
CHANGED
|
@@ -10,11 +10,11 @@ import sys
|
|
| 10 |
# lets add the local path for the audioldm_train library
|
| 11 |
sys.path.append('/content/qa-mdt')
|
| 12 |
|
| 13 |
-
from .audioldm_train.utilities.data.dataset_original_mos5 import AudioDataset as AudioDataset
|
| 14 |
-
from .audioldm_train.utilities.tools import build_dataset_json_from_list
|
| 15 |
from torch.utils.data import DataLoader
|
| 16 |
from pytorch_lightning import seed_everything
|
| 17 |
-
from .audioldm_train.utilities.tools import get_restore_step
|
| 18 |
def instantiate_from_config(config):
|
| 19 |
if not "target" in config:
|
| 20 |
if config == "__is_first_stage__":
|
|
|
|
| 10 |
# lets add the local path for the audioldm_train library
|
| 11 |
sys.path.append('/content/qa-mdt')
|
| 12 |
|
| 13 |
+
from qa_mdt.audioldm_train.utilities.data.dataset_original_mos5 import AudioDataset as AudioDataset
|
| 14 |
+
from qa_mdt.audioldm_train.utilities.tools import build_dataset_json_from_list
|
| 15 |
from torch.utils.data import DataLoader
|
| 16 |
from pytorch_lightning import seed_everything
|
| 17 |
+
from qa_mdt.audioldm_train.utilities.tools import get_restore_step
|
| 18 |
def instantiate_from_config(config):
|
| 19 |
if not "target" in config:
|
| 20 |
if config == "__is_first_stage__":
|