Update src/pipeline.py

src/pipeline.py

@@ -145,10 +145,10 @@ def load_pipeline() -> Pipeline:
     
     torch.backends.cudnn.benchmark = True
     torch.backends.cudnn.deterministic = False
-    torch.set_deterministic_debug_mode(0) 
+    # torch.set_deterministic_debug_mode(0) 
     torch.backends.cuda.matmul.allow_tf32 = True
-    torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = True
-    # torch.cuda.set_memory_growth(True)
+    # torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = True
+    torch.cuda.set_memory_growth(True)
     torch.cuda.set_per_process_memory_fraction(0.99)
     pipeline.text_encoder.to(memory_format=torch.channels_last)
     pipeline.transformer.to(memory_format=torch.channels_last)