Skip to content

Commit f257afb

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
Added updated functionality to instantiate CerebrosNotGPT from the imported and environment - agnostic class.
1 parent 4105067 commit f257afb

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

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

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -695,7 +695,11 @@ def objective(trial: optuna.Trial) -> float:
695695
max_sequence_length=MAX_SEQ_LENGTH,
696696
padding_token=tokenizer.pad_token_id
697697
)
698-
generator = CerebrosNotGPT(config)
698+
699+
700+
generator = CerebrosNotGPT(config, model_0=best_model_found)
701+
702+
raise ValueError("Debug New Model Config")
699703

700704
# mlflow.keras.log_model(generator, artifact_path="generator")
701705

0 commit comments

Comments
 (0)