ribesstefano commited on
Commit
6aa9fe5
·
1 Parent(s): ac8676d

Fixed None token to get_pipeline function

Browse files
Files changed (1) hide show
  1. protac_splitter/protac_splitter.py +1 -0
protac_splitter/protac_splitter.py CHANGED
@@ -134,6 +134,7 @@ def split_protac(
134
  if use_transformer:
135
  pipe = get_pipeline(
136
  model_name="ailab-bio/PROTAC-Splitter",
 
137
  is_causal_language_model=False,
138
  num_return_sequences=beam_size,
139
  device=device,
 
134
  if use_transformer:
135
  pipe = get_pipeline(
136
  model_name="ailab-bio/PROTAC-Splitter",
137
+ token=os.environ.get("HF_TOKEN", None),
138
  is_causal_language_model=False,
139
  num_return_sequences=beam_size,
140
  device=device,