diff --git a/configs/models/deepgen.py b/configs/models/deepgen.py
index adefe61..4f4b692 100644
--- a/configs/models/deepgen.py
+++ b/configs/models/deepgen.py
@@ -5,8 +5,8 @@
 from src.models.sd3_kontext.transformer_sd3_dynamic import SD3Transformer2DModel
 
 
-sd3_5_model_name_or_path = "model_zoo/UniPic2-SD3.5M-Kontext-2B"
-qwen2_5_vl_model_name_or_path = "model_zoo/Qwen2.5-VL-3B-Instruct"
+sd3_5_model_name_or_path = "Skywork/UniPic2-SD3.5M-Kontext-2B"
+qwen2_5_vl_model_name_or_path = "Qwen/Qwen2.5-VL-3B-Instruct"
 
 tokenizer = dict(
     type=AutoTokenizer.from_pretrained,
diff --git a/configs/models/deepgen_scb.py b/configs/models/deepgen_scb.py
index c9294de..053a07b 100644
--- a/configs/models/deepgen_scb.py
+++ b/configs/models/deepgen_scb.py
@@ -5,8 +5,8 @@
 from src.models.sd3_kontext.transformer_sd3_dynamic import SD3Transformer2DModel
 
 
-sd3_5_model_name_or_path = "model_zoo/UniPic2-SD3.5M-Kontext-2B"
-qwen2_5_vl_model_name_or_path = "model_zoo/Qwen2.5-VL-3B-Instruct"
+sd3_5_model_name_or_path = "Skywork/UniPic2-SD3.5M-Kontext-2B"
+qwen2_5_vl_model_name_or_path = "Qwen/Qwen2.5-VL-3B-Instruct"
 
 tokenizer = dict(
     type=AutoTokenizer.from_pretrained,
diff --git a/requirements.txt b/requirements.txt
index 585d0f9..93bcfca 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -37,6 +37,7 @@ huggingface-hub==0.34.4
 idna==3.10
 imageio==2.37.0
 importlib_metadata==8.6.1
+inflect==7.5.0
 ipykernel==6.29.5
 ipython==9.2.0
 ipython_pygments_lexers==1.1.1