search

54rt1n / ComfyUI-DareMerge

ComfyUI powertools for SD1.5 and SDXL model merging
GNU General Public License v3.0
67 stars 10 forks source link

Fix for error when drop_rate = 0 #13

Closed bvhari closed 3 months ago

bvhari commented 3 months ago

Without the change, drop_rate=0 will cause the creation of an IntTensor from torch.full, which causes torch.bernoulli to throw an error as it expects a FloatTensor