update
Browse files
meanaudio/model/flow_matching.py
CHANGED
|
@@ -16,7 +16,7 @@ class FlowMatching:
|
|
| 16 |
reverse_flow: bool = True):
|
| 17 |
# inference_mode: 'euler' or 'adaptive'
|
| 18 |
# num_steps: number of steps in the euler inference mode
|
| 19 |
-
# !TODO activate min_sigma
|
| 20 |
super().__init__()
|
| 21 |
self.min_sigma = min_sigma
|
| 22 |
self.inference_mode = inference_mode
|
|
|
|
| 16 |
reverse_flow: bool = True):
|
| 17 |
# inference_mode: 'euler' or 'adaptive'
|
| 18 |
# num_steps: number of steps in the euler inference mode
|
| 19 |
+
# !TODO activate min_sigma for flow matching
|
| 20 |
super().__init__()
|
| 21 |
self.min_sigma = min_sigma
|
| 22 |
self.inference_mode = inference_mode
|