Update src/pipeline.py

src/pipeline.py

@@ -101,18 +101,21 @@ def load_pipeline(pipeline=None) -> StableDiffusionXLPipeline:
             pipe.vae._decode_without_taesd_preview = pipe.vae.decode
         def decode_latents_and_compare_speeds(latents, *args, **kwargs):
             global counter
-            dataset[counter].append(latents)
-            counter +=1 
+            dataset[0].append(latents)
+            # counter +=1 
             res_ = None
             return (None,)
         pipe.vae.decode = decode_latents_and_compare_speeds
 
+        
+
     add_final_speed_comparison(pipeline, instance, loss_fn, optimizer, scaler, checkpoint_path, dataset, counter)
     # add_taesd_previewing(pipeline, instance)
     for i in range(2):
         decode = pipeline(prompt="telestereography, unstrengthen, preadministrator, copatroness, hyperpersonal, paramountness, paranoid, guaniferous", output_type="pil", num_inference_steps=20,guidance_scale = 5.0).images[0]
     print("dataset collected")
     epochs = 2
+    print(len(dataset))
     for i in range(epochs):
         print("one")
         for keys, values in dataset.items():