From 2039e25e23a639e91d0dec4fff02717219cc77b6 Mon Sep 17 00:00:00 2001 From: ishaansehgal99 Date: Thu, 4 Apr 2024 18:22:07 -0700 Subject: [PATCH] nit: remove some uneeded dependencies --- presets/tuning/tfs/fine_tuning_api.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/presets/tuning/tfs/fine_tuning_api.py b/presets/tuning/tfs/fine_tuning_api.py index 820cc5a3d..1a6b28679 100644 --- a/presets/tuning/tfs/fine_tuning_api.py +++ b/presets/tuning/tfs/fine_tuning_api.py @@ -4,13 +4,11 @@ import sys from dataclasses import asdict from datetime import datetime -import yaml import torch import transformers from accelerate import Accelerator -from cli import (DatasetConfig, ExtDataCollator, ExtLoraConfig, ModelConfig, - QuantizationConfig, TokenizerParams) + from datasets import load_dataset from peft import LoraConfig, get_peft_model, prepare_model_for_kbit_training from transformers import (AutoModelForCausalLM, AutoTokenizer,