nodes/54rt1n-ComfyUI-DareMerge
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

flipped sign

Browse Source
This commit is contained in:
Martin Bukowski
2024-01-23 14:18:20 -06:00
parent 77d1253364
commit fcde8eb66d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
components/mask_model.py
+1 -1
View File
@@ -105,7 +105,7 @@ class MagnitudeMasker:
# We can easily overrun memory with large tensors, so we chunk the tensor
delta_threshold = self.process_in_chunks(tensor=absolute_delta, threshold=threshold, **kwargs)
# Create a mask for values to keep or preserve (above the threshold)
mask = absolute_delta >= delta_threshold if invert == 'No' else absolute_delta < delta_threshold
mask = absolute_delta < delta_threshold if invert == 'No' else absolute_delta >= delta_threshold
#print(f"Delta threshold: {delta_threshold} Mask: {absolute_delta.sum()} / {absolute_delta.numel()} invert: {invert} threshold: {threshold} Above: ({mask.sum()}) Below ({(mask == False).sum()})")
return mask.view_as(model_a_param)