Browse Source

add img embedding.py

luojunhui 4 weeks ago
parent
commit
0d8e621132
1 changed files with 3 additions and 3 deletions
  1. 3 3
      applications/clip_embedding/clip_model.py

+ 3 - 3
applications/clip_embedding/clip_model.py

@@ -11,12 +11,12 @@ TRUST_REMOTE_CODE = True
 
 print(f"[model_config] Loading {MODEL_NAME} on {DEVICE} dtype={DTYPE} ...")
 
-config = AutoConfig.from_pretrained(MODEL_NAME, trust_remote_code=TRUST_REMOTE_CODE)
+config = AutoConfig.from_pretrained(pretrained_model_name_or_path=MODEL_NAME, trust_remote_code=TRUST_REMOTE_CODE)
 model = AutoModel.from_pretrained(
-    MODEL_NAME, config=config, trust_remote_code=TRUST_REMOTE_CODE
+    pretrained_model_name_or_path=MODEL_NAME, config=config, trust_remote_code=TRUST_REMOTE_CODE
 ).to(dtype=DTYPE, device=DEVICE).eval()
 
-processor = CLIPProcessor.from_pretrained(MODEL_NAME)
+processor = CLIPProcessor.from_pretrained(pretrained_model_name_or_path=MODEL_NAME, trust_remote_code=TRUST_REMOTE_CODE)
 
 def get_model():
     return model, processor, DEVICE, DTYPE, MAX_BATCH