diff --git a/llm-complete-guide/configs/embeddings.yaml b/llm-complete-guide/configs/embeddings.yaml index 90fcf673..7205a3db 100644 --- a/llm-complete-guide/configs/embeddings.yaml +++ b/llm-complete-guide/configs/embeddings.yaml @@ -26,6 +26,7 @@ settings: - rerankers[flashrank] - datasets - torch + - github environment: ZENML_PROJECT_SECRET_NAME: llm_complete diff --git a/llm-complete-guide/configs/rag_eval.yaml b/llm-complete-guide/configs/rag_eval.yaml index af218835..277fafc5 100644 --- a/llm-complete-guide/configs/rag_eval.yaml +++ b/llm-complete-guide/configs/rag_eval.yaml @@ -12,6 +12,7 @@ settings: - numpy - psycopg2-binary - tiktoken + - github # configuration of the Model Control Plane model: diff --git a/llm-complete-guide/configs/rag_gcp.yaml b/llm-complete-guide/configs/rag_gcp.yaml index 7573fb86..b6830cee 100644 --- a/llm-complete-guide/configs/rag_gcp.yaml +++ b/llm-complete-guide/configs/rag_gcp.yaml @@ -12,6 +12,7 @@ settings: - tiktoken - ratelimit - rerankers + - github environment: ZENML_PROJECT_SECRET_NAME: llm_complete ZENML_ENABLE_RICH_TRACEBACK: FALSE diff --git a/llm-complete-guide/configs/rag_local_dev.yaml b/llm-complete-guide/configs/rag_local_dev.yaml index be6a7b03..0806638e 100644 --- a/llm-complete-guide/configs/rag_local_dev.yaml +++ b/llm-complete-guide/configs/rag_local_dev.yaml @@ -14,6 +14,7 @@ settings: - tiktoken - ratelimit - rerankers + - github environment: ZENML_PROJECT_SECRET_NAME: llm_complete ZENML_ENABLE_RICH_TRACEBACK: FALSE diff --git a/llm-complete-guide/configs/synthetic.yaml b/llm-complete-guide/configs/synthetic.yaml index afade7b2..9ef166e4 100644 --- a/llm-complete-guide/configs/synthetic.yaml +++ b/llm-complete-guide/configs/synthetic.yaml @@ -24,6 +24,7 @@ settings: - datasets - torch - distilabel + - github environment: ZENML_PROJECT_SECRET_NAME: llm_complete diff --git a/llm-complete-guide/run.py b/llm-complete-guide/run.py index 39a77ae9..5e351478 100644 --- a/llm-complete-guide/run.py +++ b/llm-complete-guide/run.py @@ -208,7 +208,7 @@ def main( config_path = os.path.join( os.path.dirname(os.path.realpath(__file__)), "configs", - "rag_local_dev.yaml", + config, ) if rag: