jadechoghari commited on
Commit
0f80d94
1 Parent(s): f1b1c0c

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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 MOSDiffusionPipeline(DiffusionPipeline):
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
  """