torch dynamic profiling

2026-09-11 07:18:44 +02:00 · 2024-05-16 18:40:18 -04:00
parent d63f35e298
commit 554e5c8224
19 changed files with 45 additions and 29 deletions
@@ -109,7 +109,7 @@ def taesd_vae_encode(image):
 def vae_decode(latents, model, output_type='np', full_quality=True):
    t0 = time.time()
    prev_job = shared.state.job
-    shared.state.job = 'vae'
+    shared.state.job = 'VAE'
    if not torch.is_tensor(latents): # already decoded
        return latents
    if latents.shape[0] == 0: