diff --git a/scripts/t2v_helpers/general_utils.py b/scripts/t2v_helpers/general_utils.py index b5ed536..7848257 100644 --- a/scripts/t2v_helpers/general_utils.py +++ b/scripts/t2v_helpers/general_utils.py @@ -18,11 +18,11 @@ def get_model_location(model_name): assert model_name is not None if model_name == "": - return os.path.join(ph.models_path, 'models/ModelScope/t2v') + return os.path.join(ph.models_path, 'ModelScope/t2v') elif model_name == "": - return os.path.join(ph.models_path, 'models/VideoCrafter') + return os.path.join(ph.models_path, 'VideoCrafter') else: - return os.path.join(ph.models_path, 'models/text2video/', model_name) + return os.path.join(ph.models_path, 'text2video/', model_name) def reconstruct_conds(cond, uncond, step): c = reconstruct_cond_batch(cond, step)