diff --git a/unsloth/chat_templates.py b/unsloth/chat_templates.py index 5785894a23..29eb8618aa 100644 --- a/unsloth/chat_templates.py +++ b/unsloth/chat_templates.py @@ -1468,7 +1468,15 @@ def standardize_sharegpt( return { "conversations" : all_convos, } pass - return dataset.map(_standardize_dataset, batched = True, desc = "Standardizing format") + from multiprocessing import cpu_count + num_proc = cpu_count() + + return dataset.map( + _standardize_dataset, + batched = True, + desc = "Unsloth: Standardizing formats", + num_proc = num_proc, + ) pass diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 0cb8349b59..8b144ec1c7 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -129,8 +129,12 @@ class FastBaseModel: try: vllm_version = f" vLLM: {importlib_version('vllm')}." except: vllm_version = "" + model_name = model_types[0] + if model_name == "siglip" and len(model_types) != 1: + model_name = model_types[1] + statistics = \ - f"==((====))== Unsloth {__version__}: Fast {model_types[0].title()} patching. Transformers: {transformers_version}.{vllm_version}\n"\ + f"==((====))== Unsloth {__version__}: Fast {model_name.title()} patching. Transformers: {transformers_version}.{vllm_version}\n"\ f" {chr(92)}{chr(92)} /| {gpu_stats.name}. Num GPUs = {torch.cuda.device_count()}. Max memory: {max_memory} GB. Platform: {platform_system}.\n"\ f"O^O/ {chr(92)}_/ {chr(92)} Torch: {torch.__version__}. CUDA: {gpu_stats.major}.{gpu_stats.minor}. CUDA Toolkit: {torch.version.cuda}. Triton: {triton_version}\n"\ f"{chr(92)} / Bfloat16 = {str(SUPPORTS_BFLOAT16).upper()}. FA [Xformers = {xformers_version}. FA2 = {HAS_FLASH_ATTENTION}]\n"\