Skip to content

Commit 22fe1f7

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
Comment out other occurrence of log_model.
1 parent e48f904 commit 22fe1f7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -731,7 +731,7 @@ def reset_state(self):
731731
MODEL_FILE_NAME = "cerebros-foundation-model.keras"
732732

733733
best_model_found = cerebros_automl.get_best_model(purge_model_storage_files='slate')
734-
mlflow.keras.log_model(best_model_found, artifact_path="base")
734+
# mlflow.keras.log_model(best_model_found, artifact_path="base")
735735
# best_model_found.save(MODEL_FILE_NAME)
736736
# del(best_model_found)
737737
# del(cerebros_automl)

0 commit comments

Comments
 (0)