Remove trace tensor import

monorimet · web-flow · commit e553632e4d91 · 2024-10-21T10:08:58.000-05:00
diff --git a/src/diffusers/models/unets/unet_2d_blocks.py b/src/diffusers/models/unets/unet_2d_blocks.py
@@ -35,7 +35,6 @@
 )
 from ..transformers.dual_transformer_2d import DualTransformer2DModel
 from ..transformers.transformer_2d import Transformer2DModel
-import iree.turbine.ops.iree as iree_ops
 
 
 logger = logging.get_logger(__name__)  # pylint: disable=invalid-name

Original file line number	Diff line number	Diff line change
`@@ -35,7 +35,6 @@`
`35`	`35`	`)`
`36`	`36`	`from ..transformers.dual_transformer_2d import DualTransformer2DModel`
`37`	`37`	`from ..transformers.transformer_2d import Transformer2DModel`
`38`		`-import iree.turbine.ops.iree as iree_ops`
`39`	`38`
`40`	`39`
`41`	`40`	`logger = logging.get_logger(__name__) # pylint: disable=invalid-name`