RohitGandikota commited on
Commit
f52ac7e
Β·
1 Parent(s): 66982a6

pushing training code

Browse files
trainscripts/textsliders/demotrain.py CHANGED
@@ -13,18 +13,18 @@ from tqdm import tqdm
13
 
14
 
15
  from trainscripts.textsliders.lora import LoRANetwork, DEFAULT_TARGET_REPLACE, UNET_TARGET_REPLACE_MODULE_CONV
16
- import train_util
17
- import model_util
18
- import prompt_util
19
- from prompt_util import (
20
  PromptEmbedsCache,
21
  PromptEmbedsPair,
22
  PromptSettings,
23
  PromptEmbedsXL,
24
  )
25
- import debug_util
26
- import config_util
27
- from config_util import RootConfig
28
 
29
  import wandb
30
 
 
13
 
14
 
15
  from trainscripts.textsliders.lora import LoRANetwork, DEFAULT_TARGET_REPLACE, UNET_TARGET_REPLACE_MODULE_CONV
16
+ import trainscripts.textsliders.train_util as train_util
17
+ import trainscripts.textsliders.model_util as model_util
18
+ import trainscripts.textsliders.prompt_util as prompt_util
19
+ from trainscripts.textsliders.prompt_util import (
20
  PromptEmbedsCache,
21
  PromptEmbedsPair,
22
  PromptSettings,
23
  PromptEmbedsXL,
24
  )
25
+ import trainscripts.textsliders.debug_util as debug_util
26
+ import trainscripts.textsliders.config_util as config_util
27
+ from trainscripts.textsliders.config_util import RootConfig
28
 
29
  import wandb
30