nodes/Reithan-negative_rejection_steering
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add flux to the Prediction Type inference (#18)

Browse Source 
Testing it in Comfy, it seems like it works as `EPS`, though it also
worked with `V`.
This commit is contained in:
Alexander Brown
2025-07-31 23:23:09 -07:00
committed by GitHub
parent d26fcf6fc8
commit fc38b5c998
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
NRS/nodes_NRS.py
+2 -1
View File
@@ -13,6 +13,7 @@ class PredictionType(Enum):
_RAW_TO_ENUM = { _RAW_TO_ENUM = {
"eps": PredictionType.EPS, "eps": PredictionType.EPS,
"epsilon": PredictionType.EPS, "epsilon": PredictionType.EPS,
"flux": PredictionType.EPS,
"v": PredictionType.V, "v": PredictionType.V,
"v_prediction": PredictionType.V, "v_prediction": PredictionType.V,
"x0": PredictionType.X0, "x0": PredictionType.X0,
@@ -218,4 +219,4 @@ class NRS:
NODE_CLASS_MAPPINGS = { NODE_CLASS_MAPPINGS = {
"NRS": NRS, "NRS": NRS,
} }