jadechoghari commited on
Commit
7a073d1
1 Parent(s): 03aa894

Update generate.py

Browse files
Files changed (1) hide show
  1. generate.py +3 -3
generate.py CHANGED
@@ -5,11 +5,11 @@ from tqdm import tqdm
5
  import os
6
  from transformers import logging
7
 
8
- from .utils import CONTROLNET_DICT
9
- from .utils import load_config, save_config
10
  from .utils import get_controlnet_kwargs, get_frame_ids, get_latents_dir, init_model, seed_everything
11
  from .utils import prepare_control, load_latent, load_video, prepare_depth, save_video
12
- from .utils import register_time, register_attention_control, register_conv_control
13
 
14
  # will cause an issue
15
  from . import vidtome
 
5
  import os
6
  from transformers import logging
7
 
8
+ from .controlnet_utils import CONTROLNET_DICT
9
+ from .config_utils import load_config, save_config
10
  from .utils import get_controlnet_kwargs, get_frame_ids, get_latents_dir, init_model, seed_everything
11
  from .utils import prepare_control, load_latent, load_video, prepare_depth, save_video
12
+ from .pnp_utils import register_time, register_attention_control, register_conv_control
13
 
14
  # will cause an issue
15
  from . import vidtome