jadechoghari
commited on
Commit
•
7a073d1
1
Parent(s):
03aa894
Update generate.py
Browse files- 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 .
|
9 |
-
from .
|
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 .
|
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
|