|
|
@ -21,6 +21,8 @@ import comfy.model_management
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def cast_to(weight, dtype=None, device=None, non_blocking=False):
|
|
|
|
def cast_to(weight, dtype=None, device=None, non_blocking=False):
|
|
|
|
|
|
|
|
if (dtype is None or weight.dtype == dtype) and (device is None or weight.device == device):
|
|
|
|
|
|
|
|
return weight
|
|
|
|
r = torch.empty_like(weight, dtype=dtype, device=device)
|
|
|
|
r = torch.empty_like(weight, dtype=dtype, device=device)
|
|
|
|
r.copy_(weight, non_blocking=non_blocking)
|
|
|
|
r.copy_(weight, non_blocking=non_blocking)
|
|
|
|
return r
|
|
|
|
return r
|
|
|
|