Skip to content

Commit

Permalink
add missing mandatory parameters for generate_data
Browse files Browse the repository at this point in the history
  • Loading branch information
JudeNiroshan committed Oct 4, 2024
1 parent 1e718f4 commit 9c21c7c
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 1 deletion.
3 changes: 2 additions & 1 deletion pipeline.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -1221,7 +1221,8 @@ deploymentSpec:
\ `empty`\n # it allows generating from the whole repo, see:\n # https://github.com/instructlab/sdg/blob/c6a9e74a1618b1077cd38e713b8aaed8b7c0c8ce/src/instructlab/sdg/utils/taxonomy.py#L230\n\
\ generate_data(\n client=client,\n num_instructions_to_generate=num_instructions_to_generate,\n\
\ output_dir=sdg.path,\n taxonomy=taxonomy.path,\n \
\ taxonomy_base=taxonomy_base,\n model_name=model,\n )\n\n"
\ taxonomy_base=taxonomy_base,\n model_name=model,\n chunk_word_count=1000,\n\
\ server_ctx_size=4096\n )\n\n"
image: quay.io/tcoufal/ilab-sdg:latest
pipelineInfo:
description: InstructLab pipeline
Expand Down
2 changes: 2 additions & 0 deletions sdg/components.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,4 +60,6 @@ def sdg_op(
taxonomy=taxonomy.path,
taxonomy_base=taxonomy_base,
model_name=model,
chunk_word_count=1000,
server_ctx_size=4096,
)

0 comments on commit 9c21c7c

Please sign in to comment.