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

Another attempt at logging models by mlflow.

Author: david-thrower

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

@@ -730,7 +730,7 @@ def reset_state(self):
         MODEL_FILE_NAME = "cerebros-foundation-model.keras"
 
         best_model_found = cerebros_automl.get_best_model(purge_model_storage_files=True)
-        # mlflow.keras.log_model(best_model_found, name="base")
+        mlflow.keras.log_model(best_model_found, artifact_path="base")
         # best_model_found.save(MODEL_FILE_NAME)
         # del(best_model_found)
         # del(cerebros_automl)
@@ -881,7 +881,7 @@ def call(self, inputs):
         )
         generator = CerebrosNotGPT(config)
 
-        # mlflow.keras.log_model(generator,name="generator")
+        mlflow.keras.log_model(generator, artifact_path="generator")
         print("########### BEFORE SEARIALIZING THE GENERATIVE MODEL")
 
         def complete_text(text):