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
|