instruct model setup

This commit is contained in:
Your Name
2025-08-28 17:57:59 +00:00
parent 77c563f358
commit d49b4ff2d5
55 changed files with 27760 additions and 326 deletions
+17 -4
View File
@@ -1,7 +1,7 @@
"""
2025.8.4
2025.8.5
4.55.1
2025.8.9
2025.8.10
4.55.4
0.21.0
__UNSLOTH_VERSIONING__
"""
@@ -152,9 +152,10 @@ class UnslothDDPOConfig(DDPOConfig):
default = -1,
metadata = {'help': 'Chunk size to reduce memory usage. -1 is most efficient.'},
)
def __init__(
self,
exp_name = 'inference',
exp_name = 'train',
run_name = '',
seed = 3407,
log_with = None,
@@ -194,6 +195,7 @@ class UnslothDDPOConfig(DDPOConfig):
push_to_hub = False,
vllm_sampling_params = None,
unsloth_num_chunks = -1,
**kwargs,
):
@@ -238,6 +240,7 @@ class UnslothDDPOConfig(DDPOConfig):
push_to_hub = push_to_hub,**kwargs)
self.vllm_sampling_params = vllm_sampling_params
self.unsloth_num_chunks = unsloth_num_chunks
pass
class _UnslothDDPOTrainer(PyTorchModelHubMixin):
@@ -899,3 +902,13 @@ class UnslothDDPOTrainer(_UnslothDDPOTrainer):
image_samples_hook = image_samples_hook,**kwargs)
pass
if hasattr(logger, "addFilter"):
import logging
class HideLoggingMessage(logging.Filter):
def __init__(self, text): self.text = text
def filter(self, x): return not (self.text in x.getMessage())
pass
logger.addFilter(HideLoggingMessage("`use_cache=True`"))