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 7543b1f4e..1482d9142 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 @@ -14,24 +14,6 @@ from typing import Any, Dict, List, Optional, Tuple import torch from llama_models.sku_list import resolve_model -from torch import nn -from torch.optim import Optimizer -from torch.utils.data import DataLoader, DistributedSampler -from torchtune import modules, training, utils as torchtune_utils -from torchtune.data import padded_collate_sft - -from torchtune.modules.loss import CEWithChunkedOutputLoss -from torchtune.modules.peft import ( - get_adapter_params, - get_adapter_state_dict, - get_lora_module_names, - get_merged_lora_ckpt, - set_trainable_params, - validate_missing_and_unexpected_for_lora, -) -from torchtune.training.lr_schedulers import get_cosine_schedule_with_warmup -from torchtune.training.metric_logging import DiskLogger -from tqdm import tqdm from llama_stack.apis.common.training_types import PostTrainingMetric from llama_stack.apis.datasetio import DatasetIO