From 8d6b6836acfbc629b66536c8e0efbf3b67337f09 Mon Sep 17 00:00:00 2001 From: saienduri Date: Thu, 10 Oct 2024 01:07:49 -0500 Subject: [PATCH] namespace change --- src/diffusers/models/attention_processor.py | 2 +- src/diffusers/models/embeddings.py | 2 +- src/diffusers/models/unets/unet_2d_blocks.py | 2 +- src/diffusers/models/unets/unet_2d_condition.py | 2 +- src/diffusers/schedulers/scheduling_euler_ancestral_discrete.py | 2 +- src/diffusers/schedulers/scheduling_pndm.py | 2 +- 6 files changed, 6 insertions(+), 6 deletions(-) diff --git a/src/diffusers/models/attention_processor.py b/src/diffusers/models/attention_processor.py index d71b79ad6d51..8d03a805ba6b 100644 --- a/src/diffusers/models/attention_processor.py +++ b/src/diffusers/models/attention_processor.py @@ -26,7 +26,7 @@ from ..utils.import_utils import is_torch_npu_available, is_xformers_available from ..utils.torch_utils import maybe_allow_in_graph from .lora import LoRALinearLayer -#from shark_turbine.ops.iree import trace_tensor +#from iree.turbine.ops.iree import trace_tensor logger = logging.get_logger(__name__) # pylint: disable=invalid-name diff --git a/src/diffusers/models/embeddings.py b/src/diffusers/models/embeddings.py index bbf996da42a2..f712da572d7f 100644 --- a/src/diffusers/models/embeddings.py +++ b/src/diffusers/models/embeddings.py @@ -23,7 +23,7 @@ from .activations import FP32SiLU, get_activation from .attention_processor import Attention -from shark_turbine.ops.iree import trace_tensor +from iree.turbine.ops.iree import trace_tensor def get_timestep_embedding( diff --git a/src/diffusers/models/unets/unet_2d_blocks.py b/src/diffusers/models/unets/unet_2d_blocks.py index e2521c525923..da9f89a945e7 100644 --- a/src/diffusers/models/unets/unet_2d_blocks.py +++ b/src/diffusers/models/unets/unet_2d_blocks.py @@ -35,7 +35,7 @@ ) from ..transformers.dual_transformer_2d import DualTransformer2DModel from ..transformers.transformer_2d import Transformer2DModel -import shark_turbine.ops.iree as iree_ops +import iree.turbine.ops.iree as iree_ops logger = logging.get_logger(__name__) # pylint: disable=invalid-name diff --git a/src/diffusers/models/unets/unet_2d_condition.py b/src/diffusers/models/unets/unet_2d_condition.py index add018d246c2..294fab3960f7 100644 --- a/src/diffusers/models/unets/unet_2d_condition.py +++ b/src/diffusers/models/unets/unet_2d_condition.py @@ -49,7 +49,7 @@ get_mid_block, get_up_block, ) -import shark_turbine.ops.iree as iree_ops +import iree.turbine.ops.iree as iree_ops logger = logging.get_logger(__name__) # pylint: disable=invalid-name diff --git a/src/diffusers/schedulers/scheduling_euler_ancestral_discrete.py b/src/diffusers/schedulers/scheduling_euler_ancestral_discrete.py index 082acb0db2f3..751947b3ca31 100644 --- a/src/diffusers/schedulers/scheduling_euler_ancestral_discrete.py +++ b/src/diffusers/schedulers/scheduling_euler_ancestral_discrete.py @@ -23,7 +23,7 @@ from ..utils import BaseOutput, logging from ..utils.torch_utils import randn_tensor from .scheduling_utils import KarrasDiffusionSchedulers, SchedulerMixin -#from shark_turbine.ops.iree import trace_tensor +#from iree.turbine.ops.iree import trace_tensor logger = logging.get_logger(__name__) # pylint: disable=invalid-name diff --git a/src/diffusers/schedulers/scheduling_pndm.py b/src/diffusers/schedulers/scheduling_pndm.py index b7a0b5a41b6c..15aebfdb99be 100644 --- a/src/diffusers/schedulers/scheduling_pndm.py +++ b/src/diffusers/schedulers/scheduling_pndm.py @@ -22,7 +22,7 @@ from ..configuration_utils import ConfigMixin, register_to_config from .scheduling_utils import KarrasDiffusionSchedulers, SchedulerMixin, SchedulerOutput -#from shark_turbine.ops.iree import trace_tensor +#from iree.turbine.ops.iree import trace_tensor # Copied from diffusers.schedulers.scheduling_ddpm.betas_for_alpha_bar def betas_for_alpha_bar(