Commit
•
61de3b8
1
Parent(s):
02b5cd2
Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -71,7 +71,7 @@ class MOSDiffusionPipeline(DiffusionPipeline):
|
|
| 71 |
"""
|
| 72 |
Run the MOS Diffusion Pipeline. This method calls the infer function from infer_mos5.py.
|
| 73 |
"""
|
| 74 |
-
from infer.infer_mos5 import infer
|
| 75 |
|
| 76 |
infer(
|
| 77 |
dataset_key=self.dataset_key,
|
|
|
|
| 71 |
"""
|
| 72 |
Run the MOS Diffusion Pipeline. This method calls the infer function from infer_mos5.py.
|
| 73 |
"""
|
| 74 |
+
from .infer.infer_mos5 import infer
|
| 75 |
|
| 76 |
infer(
|
| 77 |
dataset_key=self.dataset_key,
|