From c93dee5155963b57e5564d20cc75a2cc50bceeba Mon Sep 17 00:00:00 2001 From: FuouM Date: Sun, 11 Aug 2024 19:04:54 +0700 Subject: [PATCH] fix vfi mamba import --- vfi_mamba_run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vfi_mamba_run.py b/vfi_mamba_run.py index 6613ac4..d525ad4 100644 --- a/vfi_mamba_run.py +++ b/vfi_mamba_run.py @@ -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