Fix pylint

This commit is contained in:
doctorpangloss 2024-08-26 16:56:27 -07:00
parent f49bcd4f3c
commit 27f4d70904

View File

@ -365,6 +365,7 @@ def calculate_weight(patches, weight, key, intermediate_dtype=torch.float32):
if isinstance(v, list):
v = (calculate_weight(v[1:], v[0].clone(), key, intermediate_dtype=intermediate_dtype), )
patch_type = ""
if len(v) == 1:
patch_type = "diff"
elif len(v) == 2: