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: