Merge pull request #3 from nod-ai/0.29.0.dev0-shark-fix

Adapt to shark_turbine -> iree.turbine rename

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

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(

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

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

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
 

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(