phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-06 02:32:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

temp commit

Browse source
This commit is contained in:
Botao Chen 2025-01-03 14:41:47 -08:00
parent 82d575811c
commit 280581a4a3
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/inline/post_training/torchtune/recipes/lora_finetuning_single_device.py
View file

@ -126,7 +126,8 @@ class LoraFinetuningSingleDevice:
self._output_dir = str(DEFAULT_CHECKPOINT_DIR) 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.epochs_run = 0
self.total_epochs = training_config.n_epochs self.total_epochs = training_config.n_epochs
self._data_format = training_config.data_config.data_format self._data_format = training_config.data_config.data_format
@ -365,6 +366,7 @@ class LoraFinetuningSingleDevice:
column_map=self._column_map, column_map=self._column_map,
) )
data_transform = await utils.get_data_transform(self._data_format) data_transform = await utils.get_data_transform(self._data_format)
print("data_transform", data_transform.__name__)
ds = SFTDataset( ds = SFTDataset(
rows, rows,
message_transform=data_transform( message_transform=data_transform(