fix vfi mamba import

This commit is contained in:
FuouM
2024-08-11 19:04:54 +07:00
parent 7aef6d9a2d
commit c93dee5155
+1 -1
View File
@@ -3,7 +3,7 @@ from pathlib import Path
import torch
from .constants import VFI_MAMBA_DEFAULT, VFI_MAMBA_MODELS, VFI_MAMBA_PATHS
from .vfimamba_module.vfi_mamba_model import run_vfi_mamba
from .module_vfimamba.vfi_mamba_model import run_vfi_mamba
base_dir = Path(__file__).resolve().parent