Add flux to the Prediction Type inference (#18)

Testing it in Comfy, it seems like it works as `EPS`, though it also worked with `V`.
2025-07-31 23:23:09 -07:00
parent d26fcf6fc8
commit fc38b5c998
1 changed files with 2 additions and 1 deletions
@@ -13,6 +13,7 @@ class PredictionType(Enum):
 _RAW_TO_ENUM = {
    "eps":          PredictionType.EPS,
    "epsilon":      PredictionType.EPS,
+    "flux":         PredictionType.EPS,
    "v":            PredictionType.V,
    "v_prediction": PredictionType.V,
    "x0":           PredictionType.X0,
@@ -218,4 +219,4 @@ class NRS:

 NODE_CLASS_MAPPINGS = {
    "NRS": NRS,
-}
+}