comment trace tensor

Author: monorimet

models/turbine_models/custom_models/sd3_inference/sd3_schedulers.py

@@ -66,7 +66,7 @@ def __init__(
     def initialize(self, sample):
         step_count = torch.tensor(len(self.timesteps))
         timesteps = self.model.timesteps
-        ops.trace_tensor("sample", sample[:,:,0,0])
+        #ops.trace_tensor("sample", sample[:,:,0,0])
         return (
             sample,
             step_count,