This commit is contained in:
Disty0
2023-09-08 04:17:27 +03:00
parent 9cadf4fc10
commit 12c2f4d6ee
-3
View File
@@ -1,11 +1,8 @@
import torch
import intel_extension_for_pytorch as ipex # pylint: disable=import-error, unused-import
import diffusers #0.20.2 # pylint: disable=import-error
# pylint: disable=protected-access, missing-function-docstring, line-too-long
Attention = diffusers.models.attention_processor.Attention
original_torch_bmm = torch.bmm
def torch_bmm(input, mat2, *, out=None):
if input.dtype != mat2.dtype: