diff --git a/examples/scripts/sft_video_llm.py b/examples/scripts/sft_video_llm.py index 3dd0995799..aeb79433e6 100644 --- a/examples/scripts/sft_video_llm.py +++ b/examples/scripts/sft_video_llm.py @@ -50,12 +50,12 @@ import requests import torch +import wandb from datasets import load_dataset from peft import LoraConfig from qwen_vl_utils import process_vision_info from transformers import AutoModelForVision2Seq, AutoProcessor, BitsAndBytesConfig, Qwen2VLProcessor -import wandb from trl import ModelConfig, ScriptArguments, SFTConfig, SFTTrainer, TrlParser, get_kbit_device_map