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
|