Remove trace tensor imports.

nod-ai · Oct 21, 2024 · 599dd4e · 599dd4e
1 parent aa45bc9
commit 599dd4e
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/src/diffusers/schedulers/scheduling_euler_ancestral_discrete.py b/src/diffusers/schedulers/scheduling_euler_ancestral_discrete.py
@@ -23,7 +23,6 @@
 from ..utils import BaseOutput, logging
 from ..utils.torch_utils import randn_tensor
 from .scheduling_utils import KarrasDiffusionSchedulers, SchedulerMixin
-#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
@@ -22,7 +22,6 @@
 
 from ..configuration_utils import ConfigMixin, register_to_config
 from .scheduling_utils import KarrasDiffusionSchedulers, SchedulerMixin, SchedulerOutput
-#from iree.turbine.ops.iree import trace_tensor
 
 # Copied from diffusers.schedulers.scheduling_ddpm.betas_for_alpha_bar
 def betas_for_alpha_bar(