gabrielbianchin commited on
Commit
a39c0d4
·
1 Parent(s): 4b372c3
Files changed (1) hide show
  1. regression/tokenizer_bbb.py +3 -3
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, "normalize_cls_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_cls_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_cls_text.joblib")
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)