jadechoghari
commited on
Commit
•
02b5cd2
1
Parent(s):
1752041
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_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,
|