FERMAT_TEST_REST_SERVER/GLASSES_AI_SERVER
4
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

edit: 기본모델타입 b32로 변경

Browse Source
This commit is contained in:
김재웅
2025-11-10 10:49:32 +09:00
parent 55d92878d5
commit 8c2e146136
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
main_rest/app/models.py
View File

@@ -611,7 +611,7 @@ class VectorImageSearchVitReq(BaseModel):
### [Request] vector image search vit
"""
prompt : str = Field(description='프롬프트', example='검은색 안경')
modelType : str = Field(VitModelType.l14, description='pretrained model 타입', example=VitModelType.l14)
modelType : str = Field(VitModelType.b32, description='pretrained model 타입', example=VitModelType.b32)
indexType : str = Field(VitIndexType.l2, description='인덱스 타입', example=VitIndexType.l2)
searchNum : int = Field(4, description='검색결과 이미지 갯수', example=4)
@@ -621,7 +621,7 @@ class VectorImageSearchVitInputImgReq(BaseModel):
### [Request] vector image search vit - input image
"""
inputImage : str = Field(description='base64 이미지', example='')
modelType : str = Field(VitModelType.l14, description='pretrained model 타입', example=VitModelType.l14)
modelType : str = Field(VitModelType.b32, description='pretrained model 타입', example=VitModelType.b32)
indexType : str = Field(VitIndexType.l2, description='인덱스 타입', example=VitIndexType.l2)
searchNum : int = Field(4, description='검색결과 이미지 갯수', example=4)