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
|