Patch IS_CHANGED
This commit is contained in:
+1
-1
@@ -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']
|
||||
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user