Skip to content

Commit d747692

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
Move stop point to end of phase i-a tests ...
1 parent eeb22ca commit d747692

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

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

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -695,11 +695,8 @@ def objective(trial: optuna.Trial) -> float:
695695
max_sequence_length=MAX_SEQ_LENGTH,
696696
padding_token=tokenizer.pad_token_id
697697
)
698-
699-
700698
generator = CerebrosNotGPT(config, model_0=best_model_found)
701699

702-
raise ValueError("Debug New Model Config")
703700

704701
# mlflow.keras.log_model(generator, artifact_path="generator")
705702

@@ -973,7 +970,7 @@ def test_text(test_prompt: str, max_new_tokens: int, sample_number: int, result_
973970

974971
# print(f"PROMPT number {counter}: {half_sample}; RESPONSE: {full_generated_text}")
975972

976-
973+
raise ValueError("Debug New Model Config")
977974
# del(best_model_found)
978975
# del(generator)
979976
collect()

0 commit comments

Comments
 (0)