|
@@ -1,10 +1,10 @@
|
|
|
-from applications.config import MODEL_CONFIG
|
|
|
+from applications.config import LOCAL_MODEL_CONFIG
|
|
|
|
|
|
from pymilvus import Collection, CollectionSchema, FieldSchema, DataType
|
|
|
|
|
|
|
|
|
collections = {}
|
|
|
-for model_name, cfg in MODEL_CONFIG.items():
|
|
|
+for model_name, cfg in LOCAL_MODEL_CONFIG.items():
|
|
|
col_name = model_name.replace("/", "_").replace("-", "_").lower()
|
|
|
fields = [
|
|
|
FieldSchema(name="id", dtype=DataType.INT64, is_primary=True, auto_id=True),
|