diff --git a/app/utils/get_embedding_dimension.py b/app/utils/get_embedding_dimension.py
index 0328977..a936ca3 100644
--- a/app/utils/get_embedding_dimension.py
+++ b/app/utils/get_embedding_dimension.py
@@ -4,9 +4,9 @@ from app.utils.llm_utils import create_embeddings
 # 通用测试方法，获取模型的向量维度
 def get_embedding_dimension(model, test_text="test"):
   embedding = model.embed_query(test_text)
-  print("embedding", embedding)
+  # print("embedding", embedding)
   return len(embedding) if hasattr(embedding, '__len__') else embedding.shape[-1]
 
 
-dimension = get_embedding_dimension(create_embeddings())
+dimension = get_embedding_dimension(create_embeddings("bailian-embedding"))
 print(f"模型向量维度: {dimension}")