diff --git a/library/lumina_train_util.py b/library/lumina_train_util.py
index 414b2849..db9af238 100644
--- a/library/lumina_train_util.py
+++ b/library/lumina_train_util.py
@@ -1,5 +1,4 @@
 import inspect
-import enum
 import argparse
 import math
 import os
@@ -9,20 +8,16 @@ from typing import Callable, Dict, List, Optional, Tuple, Any, Union
 
 import torch
 from torch import Tensor
-from torchdiffeq import odeint
 from accelerate import Accelerator, PartialState
 from transformers import Gemma2Model
 from tqdm import tqdm
 from PIL import Image
 from safetensors.torch import save_file
 
-from diffusers.schedulers.scheduling_heun_discrete import HeunDiscreteScheduler
-from library import lumina_models, lumina_util, strategy_base, strategy_lumina, train_util
+from library import lumina_models, strategy_base, strategy_lumina, train_util
 from library.flux_models import AutoEncoder
 from library.device_utils import init_ipex, clean_memory_on_device
 from library.sd3_train_utils import FlowMatchEulerDiscreteScheduler
-from library.lumina_dpm_solver import NoiseScheduleFlow, DPM_Solver
-import library.lumina_path as path
 
 init_ipex()