Remove trace tensor import

nod-ai · Oct 21, 2024 · c2b0215 · c2b0215
1 parent 9845df3
commit c2b0215
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/diffusers/models/embeddings.py b/src/diffusers/models/embeddings.py
@@ -23,8 +23,6 @@
 from .activations import FP32SiLU, get_activation
 from .attention_processor import Attention
 
-from iree.turbine.ops.iree import trace_tensor
-
 
 def get_timestep_embedding(
     timesteps: torch.Tensor,