Commit
·
a39c0d4
1
Parent(s):
4b372c3
update
Browse files
regression/tokenizer_bbb.py
CHANGED
|
@@ -40,11 +40,11 @@ class BBBTokenizer(PreTrainedTokenizer):
|
|
| 40 |
])
|
| 41 |
|
| 42 |
model_dir = snapshot_download("SaeedLab/TITAN-BBB", allow_patterns=["normalize_reg_tabular.joblib"])
|
| 43 |
-
transformer_tab_path = os.path.join(model_dir, "
|
| 44 |
model_dir = snapshot_download("SaeedLab/TITAN-BBB", allow_patterns=["normalize_reg_image.joblib"])
|
| 45 |
-
transformer_img_path = os.path.join(model_dir, "
|
| 46 |
model_dir = snapshot_download("SaeedLab/TITAN-BBB", allow_patterns=["normalize_reg_text.joblib"])
|
| 47 |
-
transformer_txt_path = os.path.join(model_dir, "
|
| 48 |
|
| 49 |
self.feature_transformer_tab = joblib.load(transformer_tab_path)
|
| 50 |
self.feature_transformer_img = joblib.load(transformer_img_path)
|
|
|
|
| 40 |
])
|
| 41 |
|
| 42 |
model_dir = snapshot_download("SaeedLab/TITAN-BBB", allow_patterns=["normalize_reg_tabular.joblib"])
|
| 43 |
+
transformer_tab_path = os.path.join(model_dir, "normalize_reg_tabular.joblib")
|
| 44 |
model_dir = snapshot_download("SaeedLab/TITAN-BBB", allow_patterns=["normalize_reg_image.joblib"])
|
| 45 |
+
transformer_img_path = os.path.join(model_dir, "normalize_reg_image.joblib")
|
| 46 |
model_dir = snapshot_download("SaeedLab/TITAN-BBB", allow_patterns=["normalize_reg_text.joblib"])
|
| 47 |
+
transformer_txt_path = os.path.join(model_dir, "normalize_reg_text.joblib")
|
| 48 |
|
| 49 |
self.feature_transformer_tab = joblib.load(transformer_tab_path)
|
| 50 |
self.feature_transformer_img = joblib.load(transformer_img_path)
|