|
@@ -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
|