|
|
|
@ -175,7 +175,7 @@ class SDClipModel(torch.nn.Module, ClipTokenWeightEncoder):
|
|
|
|
|
else: |
|
|
|
|
precision_scope = lambda a, b: contextlib.nullcontext(a) |
|
|
|
|
|
|
|
|
|
with precision_scope(model_management.get_autocast_device(device), torch.float32): |
|
|
|
|
with precision_scope(model_management.get_autocast_device(device), dtype=torch.float32): |
|
|
|
|
attention_mask = None |
|
|
|
|
if self.enable_attention_masks: |
|
|
|
|
attention_mask = torch.zeros_like(tokens) |
|
|
|
|