Commit
•
3288d65
1
Parent(s):
225b9d3
Update pipeline.py
Browse files- pipeline.py +2 -2
pipeline.py
CHANGED
|
@@ -2,8 +2,8 @@ from diffusers import DiffusionPipeline
|
|
| 2 |
import torch
|
| 3 |
import os
|
| 4 |
import yaml
|
| 5 |
-
from audioldm_train.utilities.tools import build_dataset_json_from_list
|
| 6 |
-
from infer_mos5 import infer # Importing the infer function
|
| 7 |
|
| 8 |
class MOSDiffusionPipeline(DiffusionPipeline):
|
| 9 |
|
|
|
|
| 2 |
import torch
|
| 3 |
import os
|
| 4 |
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 |
|