Update generate.py

generate.py

@@ -5,13 +5,14 @@ from tqdm import tqdm
 import os
 from transformers import logging
 
-from utils import CONTROLNET_DICT
-from utils import load_config, save_config
-from utils import get_controlnet_kwargs, get_frame_ids, get_latents_dir, init_model, seed_everything
-from utils import prepare_control, load_latent, load_video, prepare_depth, save_video
-from utils import register_time, register_attention_control, register_conv_control
-
-import vidtome
+from .utils import CONTROLNET_DICT
+from .utils import load_config, save_config
+from .utils import get_controlnet_kwargs, get_frame_ids, get_latents_dir, init_model, seed_everything
+from .utils import prepare_control, load_latent, load_video, prepare_depth, save_video
+from .utils import register_time, register_attention_control, register_conv_control
+
+# will cause an issue
+from . import vidtome
 
 # suppress partial model loading warning
 logging.set_verbosity_error()