PunchNFIT
commited on
Commit
·
b69d24b
1
Parent(s):
8d22a8e
Fix requirements and add trust_remote_code
Browse files- api_inference.py +1 -1
api_inference.py
CHANGED
|
@@ -14,7 +14,7 @@ print(f"🔍 Loading model from {MODEL_DIR} ...")
|
|
| 14 |
try:
|
| 15 |
config = AutoConfig.from_pretrained(MODEL_DIR)
|
| 16 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_DIR)
|
| 17 |
-
model = AutoModel.from_pretrained(MODEL_DIR, config=config)
|
| 18 |
model.eval()
|
| 19 |
print("✅ SNP Universal Embedding model loaded successfully.")
|
| 20 |
except Exception as e:
|
|
|
|
| 14 |
try:
|
| 15 |
config = AutoConfig.from_pretrained(MODEL_DIR)
|
| 16 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_DIR)
|
| 17 |
+
model = AutoModel.from_pretrained(MODEL_DIR, config=config, trust_remote_code=True)
|
| 18 |
model.eval()
|
| 19 |
print("✅ SNP Universal Embedding model loaded successfully.")
|
| 20 |
except Exception as e:
|