pushing training code

trainscripts/textsliders/config_util.py

@@ -5,7 +5,7 @@ import yaml
 from pydantic import BaseModel
 import torch
 
-from lora import TRAINING_METHODS
+from trainscripts.textsliders.lora import TRAINING_METHODS
 
 PRECISION_TYPES = Literal["fp32", "fp16", "bf16", "float32", "float16", "bfloat16"]
 NETWORK_TYPES = Literal["lierla", "c3lier"]

trainscripts/textsliders/train_util.py

@@ -5,7 +5,7 @@ import torch
 from transformers import CLIPTextModel, CLIPTokenizer
 from diffusers import UNet2DConditionModel, SchedulerMixin
 
-from model_util import SDXL_TEXT_ENCODER_TYPE
+from trainscripts.textsliders.model_util import SDXL_TEXT_ENCODER_TYPE
 
 from tqdm import tqdm