Remove unused check

This commit is contained in:
City
2024-08-30 03:40:15 +02:00
parent f40b149bc5
commit ec24e75d89
+2 -2
View File
@@ -121,10 +121,10 @@ class GGMLLayer(torch.nn.Module):
bias = None
non_blocking = comfy.model_management.device_supports_non_blocking(device)
if s.bias is not None:
bias = s.get_weight(s.bias.to(device), dtype) if hasattr(s, "get_weight") else s.bias
bias = s.get_weight(s.bias.to(device), dtype)
bias = comfy.ops.cast_to(bias, bias_dtype, device, non_blocking=non_blocking, copy=False)
weight = s.get_weight(s.weight.to(device), dtype) if hasattr(s, "get_weight") else s.weight
weight = s.get_weight(s.weight.to(device), dtype)
weight = comfy.ops.cast_to(weight, dtype, device, non_blocking=non_blocking, copy=False)
return weight, bias