dvilasuero HF staff commited on
Commit
bc52f37
1 Parent(s): c973277

pass token

Browse files
src/distilabel_dataset_generator/pipelines/sft.py CHANGED
@@ -217,6 +217,7 @@ def get_pipeline(num_turns, num_rows, system_prompt):
217
  llm=InferenceEndpointsLLM(
218
  model_id=MODEL,
219
  tokenizer_id=MODEL,
 
220
  generation_kwargs={
221
  "temperature": 0.8,
222
  "max_new_tokens": 1024
 
217
  llm=InferenceEndpointsLLM(
218
  model_id=MODEL,
219
  tokenizer_id=MODEL,
220
+ api_key=os.environ["HF_TOKEN"],
221
  generation_kwargs={
222
  "temperature": 0.8,
223
  "max_new_tokens": 1024