From 6d7bec8a374c610d31986f049e2296974471f58c Mon Sep 17 00:00:00 2001 From: rockerBOO Date: Sun, 23 Feb 2025 01:46:47 -0500 Subject: [PATCH] Remove non-used code --- library/lumina_train_util.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) 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()