()["weight"].stride()
)  # (2880, 1, 960, 320) having a stride of 1 for the 2nd dimension proves that it works  Tracing Tracing runs an example input tensor through the model and captures the operations that are performed on it as that input makes its way through the model’s layers. The executable or ScriptFunction that is returned is optimized with just-in-time compilation. To trace a UNet:   Copied import time
import torch
from diffusers impo