From 280581a4a3554ec05b04c056fb18f0b9f7f8a9fe Mon Sep 17 00:00:00 2001 From: Botao Chen Date: Fri, 3 Jan 2025 14:41:47 -0800 Subject: [PATCH] temp commit --- .../torchtune/recipes/lora_finetuning_single_device.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/llama_stack/providers/inline/post_training/torchtune/recipes/lora_finetuning_single_device.py b/llama_stack/providers/inline/post_training/torchtune/recipes/lora_finetuning_single_device.py index 6da5857f4..1bbae6d2e 100644 --- a/llama_stack/providers/inline/post_training/torchtune/recipes/lora_finetuning_single_device.py +++ b/llama_stack/providers/inline/post_training/torchtune/recipes/lora_finetuning_single_device.py @@ -126,7 +126,8 @@ class LoraFinetuningSingleDevice: self._output_dir = str(DEFAULT_CHECKPOINT_DIR) - self.seed = training.set_seed(seed=config.torch_seed) + # self.seed = training.set_seed(seed=config.torch_seed) + self.seed = 42 self.epochs_run = 0 self.total_epochs = training_config.n_epochs self._data_format = training_config.data_config.data_format @@ -365,6 +366,7 @@ class LoraFinetuningSingleDevice: column_map=self._column_map, ) data_transform = await utils.get_data_transform(self._data_format) + print("data_transform", data_transform.__name__) ds = SFTDataset( rows, message_transform=data_transform(