diff --git a/checkpoints/MossFormer2_SS_16K/.gitattributes b/checkpoints/MossFormer2_SS_16K/.gitattributes
deleted file mode 100644
index a6344aa..0000000
--- a/checkpoints/MossFormer2_SS_16K/.gitattributes
+++ /dev/null
@@ -1,35 +0,0 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
diff --git a/clearvoice_nodes.py b/clearvoice_nodes.py
index b7a0dd3..83d5ae7 100644
--- a/clearvoice_nodes.py
+++ b/clearvoice_nodes.py
@@ -15,7 +15,7 @@ sys.path.insert(0, current_dir)
 print(f"[ClearVoice Nodes] Current dir: {current_dir}")
 print(f"[ClearVoice Nodes] sys.path[0]: {sys.path[0]}")
 
-from clearvoice import ClearVoice
+from .clearvoice import ClearVoice
 
 
 class ClearVoiceModelLoader:
diff --git a/config/inference/MossFormer2_SS_16K.yaml b/config/inference/MossFormer2_SS_16K.yaml
index a0ba516..ebabc3c 100644
--- a/config/inference/MossFormer2_SS_16K.yaml
+++ b/config/inference/MossFormer2_SS_16K.yaml
@@ -4,7 +4,7 @@ use_cuda: 1 # 1 for True, 0 for False
 num_gpu: 1
 sampling_rate: 16000
 network: "MossFormer2_SS_16K"  ##network type
-checkpoint_dir: "checkpoints/MossFormer2_SS_16K"
+checkpoint_dir: "../../models/clearvoice//MossFormer2_SS_16K"
 input_path: "data/wsj0_2mix_16k_fullpath.lst"  # an input dir or input scp file
 #input_path: "/home/shengkui.zhao/DingTalk_NS/data/webrtc_test_set_20200521_16k/noisy"
 #input_path: "/mnt/nas_sg/mit_sg/shengkui.zhao/ComplexNN/audio/youtube_testset_16k/noisy_long/noisy"