Skip to content

Commit

Permalink
nit: remove some uneeded dependencies
Browse files Browse the repository at this point in the history
  • Loading branch information
ishaansehgal99 committed Apr 5, 2024
1 parent c3ba6ae commit 2039e25
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions presets/tuning/tfs/fine_tuning_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit 2039e25

Please sign in to comment.