From f3020dc54eb0f64ca27280940672385503d85ea9 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 30 Apr 2025 09:21:05 -0700 Subject: [PATCH] Update synthetic.py --- unsloth/dataprep/synthetic.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/unsloth/dataprep/synthetic.py b/unsloth/dataprep/synthetic.py index c8639e2d51..ba6a42373c 100644 --- a/unsloth/dataprep/synthetic.py +++ b/unsloth/dataprep/synthetic.py @@ -34,6 +34,7 @@ from .synthetic_configs import ( class SyntheticDataKit: def __init__( + self, model_name = "unsloth/Llama-3.1-8B-Instruct-unsloth-bnb-4bit", max_seq_length = 2048, gpu_memory_utilization = 0.9, @@ -128,6 +129,19 @@ class SyntheticDataKit: return pass + def from_pretrained( + self, + model_name = "unsloth/Llama-3.1-8B-Instruct-unsloth-bnb-4bit", + max_seq_length = 2048, + gpu_memory_utilization = 0.9, + float8_kv_cache = False, + conservativeness = 1.0, + token = None, + **kwargs, + ): + return self.__init__(*args, **kwargs) + pass + @staticmethod def check_vllm_status(): try: