From c2b02153bc71764df7b2305c8ee8c4f3a6ff3483 Mon Sep 17 00:00:00 2001
From: Ean Garvey <87458719+monorimet@users.noreply.github.com>
Date: Mon, 21 Oct 2024 10:08:22 -0500
Subject: [PATCH] Remove trace tensor import

---
 src/diffusers/models/embeddings.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/src/diffusers/models/embeddings.py b/src/diffusers/models/embeddings.py
index f712da572d7f..a95102169879 100644
--- 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,