Commit
•
0d7e2ec
1
Parent(s):
ab52fda
Update infer/infer_mos5.py
Browse files- infer/infer_mos5.py +1 -1
infer/infer_mos5.py
CHANGED
|
@@ -10,7 +10,7 @@ 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 |
|
| 15 |
from torch.utils.data import DataLoader
|
| 16 |
from pytorch_lightning import seed_everything
|
|
|
|
| 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 |
|
| 15 |
from torch.utils.data import DataLoader
|
| 16 |
from pytorch_lightning import seed_everything
|