mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-14 05:25:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Disable default weight values in unet conv2d for faster loading.

Browse Source
This commit is contained in:
comfyanonymous
2023-06-14 19:46:08 -04:00
parent 9d54066ebc
commit 21f04fe632
2 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ops.py
View File

@@ -15,3 +15,7 @@ class Linear(torch.nn.Module):
def forward(self, input):
return torch.nn.functional.linear(input, self.weight, self.bias)
class Conv2d(torch.nn.Conv2d):
def reset_parameters(self):
return None