davidberenstein1957 HF staff commited on
Commit
f3fb412
Β·
1 Parent(s): 0c58a58

fix: remove string getter from token

Browse files
src/distilabel_dataset_generator/sft.py CHANGED
@@ -121,7 +121,7 @@ MODEL = "meta-llama/Meta-Llama-3.1-70B-Instruct"
121
 
122
 
123
  def _run_pipeline(
124
- result_queue, _num_turns, _num_rows, _system_prompt, _token: OAuthToken = None
125
  ):
126
  os.environ["HF_TOKEN"] = _token.token
127
  with Pipeline(name="sft") as pipeline:
@@ -133,7 +133,7 @@ def _run_pipeline(
133
  generation_kwargs={
134
  "temperature": 0.8, # it's the best value for Llama 3.1 70B Instruct
135
  },
136
- api_key=_token.token,
137
  ),
138
  n_turns=_num_turns,
139
  num_rows=_num_rows,
 
121
 
122
 
123
  def _run_pipeline(
124
+ result_queue, _num_turns, _num_rows, _system_prompt, _token: str = None
125
  ):
126
  os.environ["HF_TOKEN"] = _token.token
127
  with Pipeline(name="sft") as pipeline:
 
133
  generation_kwargs={
134
  "temperature": 0.8, # it's the best value for Llama 3.1 70B Instruct
135
  },
136
+ api_key=_token,
137
  ),
138
  n_turns=_num_turns,
139
  num_rows=_num_rows,