diff --git a/requirements/framework.txt b/requirements/framework.txt
index 3019164..2085b6c 100644
--- a/requirements/framework.txt
+++ b/requirements/framework.txt
@@ -2,7 +2,7 @@ albumentations
 bezier
 einops
 modelscope
-ms-swift>=1.5.2
+ms-swift>=2.0.1
 numpy
 open_clip_torch
 opencv-python
diff --git a/scepter/modules/inference/diffusion_inference.py b/scepter/modules/inference/diffusion_inference.py
index 48fd97d..31f767f 100644
--- a/scepter/modules/inference/diffusion_inference.py
+++ b/scepter/modules/inference/diffusion_inference.py
@@ -16,7 +16,7 @@ from scepter.modules.model.registry import (BACKBONES, EMBEDDERS, MODELS,
 from scepter.modules.utils.distribute import we
 from scepter.modules.utils.file_system import FS
 
-from ...studio.utils.env import get_available_memory
+from scepter.studio.utils.env import get_available_memory
 from .control_inference import ControlInference
 from .tuner_inference import TunerInference