Fix potential issues.

This commit is contained in:
comfyanonymous
2023-11-16 14:59:54 -05:00
parent bd07ad1861
commit 9f00a18095
2 changed files with 2 additions and 2 deletions

View File

@@ -258,7 +258,7 @@ def set_attr(obj, attr, value):
for name in attrs[:-1]:
obj = getattr(obj, name)
prev = getattr(obj, attrs[-1])
setattr(obj, attrs[-1], torch.nn.Parameter(value))
setattr(obj, attrs[-1], torch.nn.Parameter(value, requires_grad=False))
del prev
def copy_to_param(obj, attr, value):