nodes/Kosinkadink-ComfyUI-AnimateDiff-Evolved
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix node args

Browse Source
This commit is contained in:
drhead
2025-01-21 11:13:31 -05:00
committed by GitHub
parent cb4d450f45
commit ba86b612e7
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
animatediff/sample_settings.py
+3 -2
View File
@@ -171,7 +171,8 @@ class NoiseLayerAdd(NoiseLayer):
class NoiseLayerAddWeighted(NoiseLayerAdd):
def __init__(self, noise_type: str, batch_offset: int, seed_gen_override: str, seed_offset: int, seed_override: int=None, mask: Tensor=None,
noise_weight=1.0, balance_multiplier=1.0):
super().__init__(noise_type, batch_offset, seed_gen_override, seed_offset, seed_override, mask, noise_weight)
super().__init__(noise_type, batch_offset, seed_gen_override, seed_offset, seed_override, mask)
self.noise_weight = noise_weight
self.balance_multiplier = balance_multiplier
self.application = NoiseApplication.ADD_WEIGHTED
@@ -183,7 +184,7 @@ class NoiseLayerAddWeighted(NoiseLayerAdd):
class NoiseLayerNormalizedSum(NoiseLayer):
def __init__(self, noise_type: str, batch_offset: int, seed_gen_override: str, seed_offset: int, seed_override: int=None, mask: Tensor=None,
noise_weight=1.0):
super().__init__(noise_type, batch_offset, seed_gen_override, seed_offset, seed_override, mask, noise_weight)
super().__init__(noise_type, batch_offset, seed_gen_override, seed_offset, seed_override, mask)
self.noise_weight = noise_weight
self.application = NoiseApplication.NORMALIZED_SUM