aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat design相关模型迁移4090测试

Browse Source 
fix
This commit is contained in:
zhouchengrong
2024-11-08 14:35:23 +08:00
parent 82c23717b0
commit 696daea775
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
app/core/config.py
View File

@@ -135,7 +135,8 @@ SEGMENTATION = {
"input": "seg_input__0", "input": "seg_input__0",
"output": "seg_output__0", "output": "seg_output__0",
} }
# ollama config
OLLAMA_URL = "http://10.1.1.243:11434/api/embeddings"
# DESIGN config # DESIGN config
DESIGN_MODEL_URL = '10.1.1.243:10000' DESIGN_MODEL_URL = '10.1.1.243:10000'
AIDA_CLOTHING = "aida-clothing" AIDA_CLOTHING = "aida-clothing"

4
app/service/search_image_with_text/service.py
View File

@@ -6,6 +6,8 @@ from chromadb.config import Settings
from chromadb.utils.embedding_functions.ollama_embedding_function import OllamaEmbeddingFunction from chromadb.utils.embedding_functions.ollama_embedding_function import OllamaEmbeddingFunction
from tqdm import tqdm from tqdm import tqdm
from app.core.config import OLLAMA_URL
# 读取 csv 文件 # 读取 csv 文件
# csv_file_path = r'D:/Files/csv/output/output.csv' # csv_file_path = r'D:/Files/csv/output/output.csv'
# image_path = r'D:/images-clean' # image_path = r'D:/images-clean'
@@ -18,7 +20,7 @@ client = chromadb.Client(Settings(is_persistent=True, persist_directory="/vector
# client = chromadb.Client(Settings(is_persistent=True, persist_directory="D:/workspace/AiDLab/vector_db")) # client = chromadb.Client(Settings(is_persistent=True, persist_directory="D:/workspace/AiDLab/vector_db"))
# 创建集合 # 创建集合
# embedding_fn = OllamaEmbeddingFunction(url="http://localhost:11434/api/embeddings", model_name="mxbai-embed-large") # embedding_fn = OllamaEmbeddingFunction(url="http://localhost:11434/api/embeddings", model_name="mxbai-embed-large")
embedding_fn = OllamaEmbeddingFunction(url="http://10.1.1.243:11434/api/embeddings", model_name="mxbai-embed-large") embedding_fn = OllamaEmbeddingFunction(url=OLLAMA_URL, model_name="mxbai-embed-large")
# def create_collection(): # def create_collection():