Update src/pipeline.py

src/pipeline.py

@@ -23,6 +23,7 @@ def error_handler(func: Callable):
             return func(*args, **kwargs)
         except Exception as e:
             print(f"Error in {func.__name__}: {str(e)}")
+            return None
     return wrapper
 
 class TorchOptimizer:
@@ -107,8 +108,10 @@ class PipelineManager:
         pipe.to("cuda")
 
         # Optimize pipeline
-        pipe = self.optimize_pipeline(pipe)
-
+        pipe_ops = self.optimize_pipeline(pipe)
+        if pipe_ops!=None:
+            pipe = pipe_ops
+            
         # Trigger compilation
         print("Running torch compilation...")
         pipe(