Update generative-proof-of-concept-CPU-preprocessing-in-memory.py

david-thrower · web-flow · commit e48f90432134 · 2025-09-23T17:36:32.000-04:00
Comment out log_model. This takes forever and is not neded for CICD scale tests.
diff --git a/generative-proof-of-concept-CPU-preprocessing-in-memory.py b/generative-proof-of-concept-CPU-preprocessing-in-memory.py
@@ -882,7 +882,7 @@ def call(self, inputs):
         )
         generator = CerebrosNotGPT(config)
         
-        mlflow.keras.log_model(generator, artifact_path="generator")
+        # mlflow.keras.log_model(generator, artifact_path="generator")
         print("########### BEFORE SEARIALIZING THE GENERATIVE MODEL")
         
         def complete_text(text):

Original file line number	Diff line number	Diff line change
`@@ -882,7 +882,7 @@ def call(self, inputs):`
`882`	`882`	`)`
`883`	`883`	`generator = CerebrosNotGPT(config)`
`884`	`884`
`885`		`- mlflow.keras.log_model(generator, artifact_path="generator")`
	`885`	`+ # mlflow.keras.log_model(generator, artifact_path="generator")`
`886`	`886`	`print("########### BEFORE SEARIALIZING THE GENERATIVE MODEL")`
`887`	`887`
`888`	`888`	`def complete_text(text):`