diff --git a/docs/snippets/technical-reference/pipeline/pipe_2.py b/docs/snippets/technical-reference/pipeline/pipe_2.py index 639ec2389..7ea99c0e0 100644 --- a/docs/snippets/technical-reference/pipeline/pipe_2.py +++ b/docs/snippets/technical-reference/pipeline/pipe_2.py @@ -10,6 +10,6 @@ instruction_dataset, num_generations=2, batch_size=1, - enable_checkpoints=True, + checkpoint_strategy=True, display_progress_bar=True, ) diff --git a/docs/tutorials/pipeline-notus-instructions-preferences-legal.ipynb b/docs/tutorials/pipeline-notus-instructions-preferences-legal.ipynb index 4c4ed36ea..58c150b3f 100644 --- a/docs/tutorials/pipeline-notus-instructions-preferences-legal.ipynb +++ b/docs/tutorials/pipeline-notus-instructions-preferences-legal.ipynb @@ -714,7 +714,7 @@ " instructions_dataset, # type: ignore\n", " num_generations=2,\n", " batch_size=8,\n", - " enable_checkpoints=True,\n", + " checkpoint_strategy=True,\n", " display_progress_bar=True,\n", ")" ] diff --git a/examples/pipeline-accelerate-and-openai.py b/examples/pipeline-accelerate-and-openai.py index f4d5a077e..0935ae262 100644 --- a/examples/pipeline-accelerate-and-openai.py +++ b/examples/pipeline-accelerate-and-openai.py @@ -76,7 +76,7 @@ def get_current_device() -> int: inputs, # type: ignore num_generations=2, batch_size=1, - enable_checkpoints=True, + checkpoint_strategy=True, display_progress_bar=True, ) dataset = gather_object(dataset) diff --git a/examples/pipeline-fn-ultrafeedback-labeller.py b/examples/pipeline-fn-ultrafeedback-labeller.py index c45721e5d..42f795159 100644 --- a/examples/pipeline-fn-ultrafeedback-labeller.py +++ b/examples/pipeline-fn-ultrafeedback-labeller.py @@ -41,7 +41,7 @@ num_generations=2, batch_size=1, display_progress_bar=True, - enable_checkpoints=True, + checkpoint_strategy=True, ) end = time.time() print("Elapsed", end - start) diff --git a/examples/pipeline-fn-ultrafeedback.py b/examples/pipeline-fn-ultrafeedback.py index 28ea1c937..b270b4b3a 100644 --- a/examples/pipeline-fn-ultrafeedback.py +++ b/examples/pipeline-fn-ultrafeedback.py @@ -50,7 +50,7 @@ dataset, # type: ignore num_generations=2, batch_size=1, - enable_checkpoints=True, + checkpoint_strategy=True, display_progress_bar=True, ) end = time.time() diff --git a/examples/pipeline-llamacpp-and-openai-process.py b/examples/pipeline-llamacpp-and-openai-process.py index 10102d08a..0a1ab1454 100644 --- a/examples/pipeline-llamacpp-and-openai-process.py +++ b/examples/pipeline-llamacpp-and-openai-process.py @@ -71,7 +71,7 @@ def load_openai_llm(task: "Task") -> "LLM": dataset, # type: ignore num_generations=2, batch_size=1, - enable_checkpoints=True, + checkpoint_strategy=True, display_progress_bar=False, ) diff --git a/examples/pipeline-llamacpp-and-openai.py b/examples/pipeline-llamacpp-and-openai.py index 3f609b012..c1c8415f6 100644 --- a/examples/pipeline-llamacpp-and-openai.py +++ b/examples/pipeline-llamacpp-and-openai.py @@ -51,7 +51,7 @@ dataset, # type: ignore num_generations=2, batch_size=1, - enable_checkpoints=True, + checkpoint_strategy=True, display_progress_bar=True, ) diff --git a/examples/pipeline-transformers-and-openai.py b/examples/pipeline-transformers-and-openai.py index a7f7761e8..7d6f3f189 100644 --- a/examples/pipeline-transformers-and-openai.py +++ b/examples/pipeline-transformers-and-openai.py @@ -58,7 +58,7 @@ dataset, # type: ignore num_generations=2, batch_size=1, - enable_checkpoints=True, + checkpoint_strategy=True, display_progress_bar=True, ) diff --git a/examples/pipeline-vllm-and-openai.py b/examples/pipeline-vllm-and-openai.py index e9f49b69e..ee042aecd 100644 --- a/examples/pipeline-vllm-and-openai.py +++ b/examples/pipeline-vllm-and-openai.py @@ -49,7 +49,7 @@ dataset, # type: ignore num_generations=2, batch_size=1, - enable_checkpoints=True, + checkpoint_strategy=True, display_progress_bar=True, )