Update src/pipeline.py
Browse files- src/pipeline.py +5 -0
src/pipeline.py
CHANGED
|
@@ -53,6 +53,11 @@ class PipelineManager:
|
|
| 53 |
# Initialize torch settings
|
| 54 |
self.optimizer.optimize_settings()
|
| 55 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 56 |
|
| 57 |
def load_transformer(self):
|
| 58 |
transformer_path = os.path.join(
|
|
|
|
| 53 |
# Initialize torch settings
|
| 54 |
self.optimizer.optimize_settings()
|
| 55 |
|
| 56 |
+
# Pre-load the pipeline during initialization
|
| 57 |
+
print("Initializing pipeline...")
|
| 58 |
+
self.pipeline = self.load_pipeline()
|
| 59 |
+
print("Pipeline initialization complete.")
|
| 60 |
+
|
| 61 |
|
| 62 |
def load_transformer(self):
|
| 63 |
transformer_path = os.path.join(
|