Experimente o profiler do tinygrad com VIZ=1 (isto é para perfilar a difusão estável). Basta `VIZ=1` e ele carrega num navegador, sem necessidade de configuração ou importações.