Patch IS_CHANGED

This commit is contained in:
Jordan Thompson
2023-08-11 00:42:05 -07:00
parent 6452a301b1
commit 41a08f6948
2 changed files with 11 additions and 10 deletions
+1 -1
View File
@@ -5,7 +5,7 @@ NODE_CLASS_MAPPINGS = {
}
NODE_DISPLAY_NAME_MAPPINGS = {
"Preset_Model_Merge": "MergeModelPreset"
"Preset_Model_Merge": "ModelMergeByPreset"
}
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
+10 -9
View File
@@ -1,5 +1,6 @@
import math
import random
import hashlib
from comfy_extras.nodes_model_merging import ModelMergeBlocks
@@ -143,16 +144,16 @@ class Preset_Model_Merge:
}
@classmethod
def IS_CHANGED(cls, **kwargs):
if kwargs.__contains__('preset'):
if kwargs['preset'] == 'RANDOM':
return float("NaN")
else:
sha256_hash = hashlib.sha256()
sha256_hash.update(kwargs['preset'].encode('utf-8'))
return sha256_hash.hexdigest()
def IS_CHANGED(cls, preset):
print("Checking change state...")
if preset == 'RANDOM':
print("IS_CHANGED == True")
return float("NaN")
else:
return 0
print("IS_CHANGED == False")
sha256_hash = hashlib.sha256()
sha256_hash.update(kwargs['preset'].encode('utf-8'))
return sha256_hash.digest().hex()
RETURN_TYPES = ("MODEL",)
FUNCTION = "merge"