jadechoghari
commited on
Commit
•
0f80d94
1
Parent(s):
f1b1c0c
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -5,7 +5,7 @@ import yaml
|
|
5 |
from .audioldm_train.utilities.tools import build_dataset_json_from_list
|
6 |
from .infer.infer_mos5 import infer # Importing the infer function
|
7 |
|
8 |
-
class
|
9 |
|
10 |
def __init__(self, config_yaml, list_inference, reload_from_ckpt=None):
|
11 |
"""
|
|
|
5 |
from .audioldm_train.utilities.tools import build_dataset_json_from_list
|
6 |
from .infer.infer_mos5 import infer # Importing the infer function
|
7 |
|
8 |
+
class QAMDTModel(DiffusionPipeline):
|
9 |
|
10 |
def __init__(self, config_yaml, list_inference, reload_from_ckpt=None):
|
11 |
"""
|