Spaces:
Runtime error
Runtime error
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:
|
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
|
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,
|