Update src/pipeline.py

src/pipeline.py

@@ -83,7 +83,7 @@ def infer(request: TextToImageRequest, pipeline: Pipeline) -> Image:
     if sample:
         clear()
         sample = None
-    torch.cuda.reset_peak_memory_stats()
+    # torch.cuda.reset_peak_memory_stats()
     generator = Generator("cuda").manual_seed(request.seed)
     image=pipeline(request.prompt,generator=generator, guidance_scale=0.0, num_inference_steps=4, max_sequence_length=256, height=request.height, width=request.width, output_type="pil").images[0]
     return(image)