jadechoghari
commited on
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 |
"""
|