tribunal-kit 9.2.3 → 9.2.4
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/.agent/.tribunal/session.jsonl +15 -0
- package/.agent/ARCHITECTURE.md +4 -4
- package/.agent/agents/agent-shield-auditor.md +18 -0
- package/.agent/agents/harness-manager.md +49 -0
- package/.agent/agents/human-gate-negotiator.md +22 -0
- package/.agent/agents/kernel-scheduler.md +56 -0
- package/.agent/agents/memory-archivist.md +37 -0
- package/.agent/agents/orchestrator.md +5 -0
- package/.agent/agents/project-planner.md +5 -0
- package/.agent/agents/security-auditor.md +5 -0
- package/.agent/rules/GEMINI.md +1 -1
- package/.agent/scripts/_utils.js +1 -1
- package/.agent/scripts/checklist.js +1 -1
- package/.agent/scripts/context_compiler.js +1 -1
- package/.agent/scripts/guardrail_engine.js +1 -1
- package/.agent/scripts/harness_manager.js +60 -0
- package/.agent/scripts/lint_runner.js +1 -1
- package/.agent/scripts/mcp_server.js +181 -0
- package/.agent/scripts/memory_archivist.js +97 -0
- package/.agent/scripts/memory_consolidator.js +37 -0
- package/.agent/scripts/memory_engine.js +61 -21
- package/.agent/scripts/pipeline_engine.js +17 -2
- package/.agent/scripts/session_logger.js +115 -0
- package/.agent/scripts/session_manager.js +10 -0
- package/.agent/scripts/swarm_dispatcher.js +74 -1
- package/.agent/scripts/syscall_registry.js +105 -0
- package/.agent/scripts/test_runner.js +1 -1
- package/.agent/scripts/verify_all.js +2 -2
- package/.agent/skill_topic_map.json +28 -1
- package/.agent/skills/agent-syscall-guidelines/SKILL.md +25 -0
- package/.agent/skills/agentic-workflows-2026/SKILL.md +8 -0
- package/.agent/skills/agentshield-security/SKILL.md +26 -0
- package/.agent/skills/context-engineering-pro/SKILL.md +7 -0
- package/.agent/skills/fabel-protocol/SKILL.md +5 -0
- package/.agent/skills/graph-engineering/SKILL.md +28 -0
- package/.agent/skills/harness-protocol/SKILL.md +8 -0
- package/.agent/skills/hf-cli/SKILL.md +239 -0
- package/.agent/skills/hf-cloud-aws-context-discovery/SKILL.md +74 -0
- package/.agent/skills/hf-cloud-python-env-setup/SKILL.md +90 -0
- package/.agent/skills/hf-cloud-python-env-setup/requirements.txt +17 -0
- package/.agent/skills/hf-cloud-python-env-setup/scripts/check_versions.py +29 -0
- package/.agent/skills/hf-cloud-python-env-setup/scripts/setup_env.py +143 -0
- package/.agent/skills/hf-cloud-sagemaker-deployment-planner/SKILL.md +90 -0
- package/.agent/skills/hf-cloud-sagemaker-iam-preflight/SKILL.md +102 -0
- package/.agent/skills/hf-cloud-sagemaker-iam-preflight/references/minimum-permissions.json +47 -0
- package/.agent/skills/hf-cloud-sagemaker-iam-preflight/references/trust-policy.json +12 -0
- package/.agent/skills/hf-cloud-sagemaker-iam-preflight/scripts/check_role.py +184 -0
- package/.agent/skills/hf-cloud-sagemaker-iam-preflight/scripts/create_role.py +118 -0
- package/.agent/skills/hf-cloud-sagemaker-production-defaults/SKILL.md +419 -0
- package/.agent/skills/hf-cloud-sagemaker-production-defaults/references/deployment-template.md +59 -0
- package/.agent/skills/hf-cloud-sagemaker-production-defaults/scripts/_common.py +148 -0
- package/.agent/skills/hf-cloud-sagemaker-production-defaults/scripts/deploy.py +327 -0
- package/.agent/skills/hf-cloud-sagemaker-production-defaults/scripts/deploy_async.py +434 -0
- package/.agent/skills/hf-cloud-sagemaker-production-defaults/scripts/deploy_ic.py +506 -0
- package/.agent/skills/hf-cloud-sagemaker-production-defaults/scripts/invoke_endpoint.py +180 -0
- package/.agent/skills/hf-cloud-sagemaker-production-defaults/scripts/teardown.py +246 -0
- package/.agent/skills/hf-cloud-serving-image-selection/SKILL.md +220 -0
- package/.agent/skills/hf-cloud-serving-image-selection/references/model-to-image.md +111 -0
- package/.agent/skills/hf-cloud-serving-image-selection/scripts/mirror_image.py +145 -0
- package/.agent/skills/hf-mem/SKILL.md +79 -0
- package/.agent/skills/huggingface-best/SKILL.md +134 -0
- package/.agent/skills/huggingface-community-evals/SKILL.md +207 -0
- package/.agent/skills/huggingface-community-evals/examples/.env.example +3 -0
- package/.agent/skills/huggingface-community-evals/examples/USAGE_EXAMPLES.md +101 -0
- package/.agent/skills/huggingface-community-evals/scripts/inspect_eval_uv.py +104 -0
- package/.agent/skills/huggingface-community-evals/scripts/inspect_vllm_uv.py +306 -0
- package/.agent/skills/huggingface-community-evals/scripts/lighteval_vllm_uv.py +297 -0
- package/.agent/skills/huggingface-datasets/SKILL.md +107 -0
- package/.agent/skills/huggingface-gradio/SKILL.md +298 -0
- package/.agent/skills/huggingface-gradio/examples.md +613 -0
- package/.agent/skills/huggingface-llm-trainer/SKILL.md +738 -0
- package/.agent/skills/huggingface-llm-trainer/references/gguf_conversion.md +296 -0
- package/.agent/skills/huggingface-llm-trainer/references/hardware_guide.md +283 -0
- package/.agent/skills/huggingface-llm-trainer/references/hub_saving.md +364 -0
- package/.agent/skills/huggingface-llm-trainer/references/local_training_macos.md +231 -0
- package/.agent/skills/huggingface-llm-trainer/references/reliability_principles.md +371 -0
- package/.agent/skills/huggingface-llm-trainer/references/trackio_guide.md +191 -0
- package/.agent/skills/huggingface-llm-trainer/references/training_methods.md +150 -0
- package/.agent/skills/huggingface-llm-trainer/references/training_patterns.md +203 -0
- package/.agent/skills/huggingface-llm-trainer/references/troubleshooting.md +282 -0
- package/.agent/skills/huggingface-llm-trainer/references/unsloth.md +313 -0
- package/.agent/skills/huggingface-llm-trainer/scripts/convert_to_gguf.py +424 -0
- package/.agent/skills/huggingface-llm-trainer/scripts/dataset_inspector.py +417 -0
- package/.agent/skills/huggingface-llm-trainer/scripts/estimate_cost.py +150 -0
- package/.agent/skills/huggingface-llm-trainer/scripts/hf_benchmarks.py +659 -0
- package/.agent/skills/huggingface-llm-trainer/scripts/train_dpo_example.py +106 -0
- package/.agent/skills/huggingface-llm-trainer/scripts/train_grpo_example.py +89 -0
- package/.agent/skills/huggingface-llm-trainer/scripts/train_sft_example.py +122 -0
- package/.agent/skills/huggingface-llm-trainer/scripts/unsloth_sft_example.py +512 -0
- package/.agent/skills/huggingface-local-models/SKILL.md +113 -0
- package/.agent/skills/huggingface-local-models/references/hardware.md +38 -0
- package/.agent/skills/huggingface-local-models/references/hub-discovery.md +178 -0
- package/.agent/skills/huggingface-local-models/references/quantization.md +256 -0
- package/.agent/skills/huggingface-lora-space-builder/SKILL.md +394 -0
- package/.agent/skills/huggingface-lora-space-builder/references/adapting-to-the-lora.md +113 -0
- package/.agent/skills/huggingface-lora-space-builder/references/base-models/krea-2.md +63 -0
- package/.agent/skills/huggingface-lora-space-builder/references/base-models/ltx.md +205 -0
- package/.agent/skills/huggingface-lora-space-builder/references/base-models/qwen-image.md +144 -0
- package/.agent/skills/huggingface-lora-space-builder/references/creative-mode.md +262 -0
- package/.agent/skills/huggingface-lora-space-builder/references/tasks.md +124 -0
- package/.agent/skills/huggingface-lora-space-builder/references/zerogpu-and-publishing.md +196 -0
- package/.agent/skills/huggingface-paper-publisher/SKILL.md +624 -0
- package/.agent/skills/huggingface-paper-publisher/examples/example_usage.md +326 -0
- package/.agent/skills/huggingface-paper-publisher/references/quick_reference.md +216 -0
- package/.agent/skills/huggingface-paper-publisher/scripts/paper_manager.py +606 -0
- package/.agent/skills/huggingface-paper-publisher/templates/arxiv.md +299 -0
- package/.agent/skills/huggingface-paper-publisher/templates/ml-report.md +358 -0
- package/.agent/skills/huggingface-paper-publisher/templates/modern.md +319 -0
- package/.agent/skills/huggingface-paper-publisher/templates/standard.md +201 -0
- package/.agent/skills/huggingface-papers/SKILL.md +239 -0
- package/.agent/skills/huggingface-spaces/README.md +9 -0
- package/.agent/skills/huggingface-spaces/SKILL.md +242 -0
- package/.agent/skills/huggingface-spaces/references/3d-cuda-extensions.md +109 -0
- package/.agent/skills/huggingface-spaces/references/3d-generation.md +98 -0
- package/.agent/skills/huggingface-spaces/references/3d-gsplat.md +53 -0
- package/.agent/skills/huggingface-spaces/references/3d-models.md +227 -0
- package/.agent/skills/huggingface-spaces/references/3d-outputs.md +114 -0
- package/.agent/skills/huggingface-spaces/references/buckets.md +89 -0
- package/.agent/skills/huggingface-spaces/references/debugging.md +236 -0
- package/.agent/skills/huggingface-spaces/references/gradio.md +200 -0
- package/.agent/skills/huggingface-spaces/references/grants.md +63 -0
- package/.agent/skills/huggingface-spaces/references/inference-providers.md +97 -0
- package/.agent/skills/huggingface-spaces/references/known-errors.md +232 -0
- package/.agent/skills/huggingface-spaces/references/requirements.md +169 -0
- package/.agent/skills/huggingface-spaces/references/zerogpu.md +349 -0
- package/.agent/skills/huggingface-tool-builder/SKILL.md +120 -0
- package/.agent/skills/huggingface-tool-builder/references/baseline_hf_api.py +57 -0
- package/.agent/skills/huggingface-tool-builder/references/baseline_hf_api.sh +40 -0
- package/.agent/skills/huggingface-tool-builder/references/baseline_hf_api.tsx +57 -0
- package/.agent/skills/huggingface-tool-builder/references/find_models_by_paper.sh +230 -0
- package/.agent/skills/huggingface-tool-builder/references/hf_enrich_models.sh +96 -0
- package/.agent/skills/huggingface-tool-builder/references/hf_model_card_frontmatter.sh +188 -0
- package/.agent/skills/huggingface-tool-builder/references/hf_model_papers_auth.sh +171 -0
- package/.agent/skills/huggingface-trackio/SKILL.md +142 -0
- package/.agent/skills/huggingface-trackio/references/alerts.md +196 -0
- package/.agent/skills/huggingface-trackio/references/logging_metrics.md +212 -0
- package/.agent/skills/huggingface-trackio/references/retrieving_metrics.md +251 -0
- package/.agent/skills/huggingface-vision-trainer/SKILL.md +593 -0
- package/.agent/skills/huggingface-vision-trainer/references/finetune_sam2_trainer.md +254 -0
- package/.agent/skills/huggingface-vision-trainer/references/hub_saving.md +618 -0
- package/.agent/skills/huggingface-vision-trainer/references/image_classification_training_notebook.md +279 -0
- package/.agent/skills/huggingface-vision-trainer/references/object_detection_training_notebook.md +700 -0
- package/.agent/skills/huggingface-vision-trainer/references/reliability_principles.md +310 -0
- package/.agent/skills/huggingface-vision-trainer/references/timm_trainer.md +91 -0
- package/.agent/skills/huggingface-vision-trainer/scripts/dataset_inspector.py +814 -0
- package/.agent/skills/huggingface-vision-trainer/scripts/estimate_cost.py +217 -0
- package/.agent/skills/huggingface-vision-trainer/scripts/image_classification_training.py +383 -0
- package/.agent/skills/huggingface-vision-trainer/scripts/object_detection_training.py +710 -0
- package/.agent/skills/huggingface-vision-trainer/scripts/sam_segmentation_training.py +382 -0
- package/.agent/skills/huggingface-zerogpu/SKILL.md +289 -0
- package/.agent/skills/huggingface-zerogpu/references/concurrency.md +79 -0
- package/.agent/skills/huggingface-zerogpu/references/cuda-and-deps.md +66 -0
- package/.agent/skills/huggingface-zerogpu/references/how-quota-works.md +74 -0
- package/.agent/skills/huggingface-zerogpu/references/how-zerogpu-works.md +50 -0
- package/.agent/skills/human-gate-negotiation/SKILL.md +30 -0
- package/.agent/skills/intelligent-routing/SKILL.md +10 -4
- package/.agent/skills/semantic-filesystem-navigation/SKILL.md +24 -0
- package/.agent/skills/session-log-interrogation/SKILL.md +29 -0
- package/.agent/skills/subagent-driven-development/SKILL.md +32 -0
- package/.agent/skills/systematic-debugging/SKILL.md +5 -0
- package/.agent/skills/train-sentence-transformers/SKILL.md +109 -0
- package/.agent/skills/train-sentence-transformers/references/base_model_selection.md +79 -0
- package/.agent/skills/train-sentence-transformers/references/dataset_formats.md +128 -0
- package/.agent/skills/train-sentence-transformers/references/evaluators_cross_encoder.md +116 -0
- package/.agent/skills/train-sentence-transformers/references/evaluators_multi_vector_encoder.md +101 -0
- package/.agent/skills/train-sentence-transformers/references/evaluators_sentence_transformer.md +151 -0
- package/.agent/skills/train-sentence-transformers/references/evaluators_sparse_encoder.md +121 -0
- package/.agent/skills/train-sentence-transformers/references/hardware_guide.md +105 -0
- package/.agent/skills/train-sentence-transformers/references/hf_jobs_execution.md +173 -0
- package/.agent/skills/train-sentence-transformers/references/losses_cross_encoder.md +137 -0
- package/.agent/skills/train-sentence-transformers/references/losses_multi_vector_encoder.md +100 -0
- package/.agent/skills/train-sentence-transformers/references/losses_sentence_transformer.md +246 -0
- package/.agent/skills/train-sentence-transformers/references/losses_sparse_encoder.md +106 -0
- package/.agent/skills/train-sentence-transformers/references/model_architectures.md +178 -0
- package/.agent/skills/train-sentence-transformers/references/prompts_and_instructions.md +160 -0
- package/.agent/skills/train-sentence-transformers/references/training_args.md +270 -0
- package/.agent/skills/train-sentence-transformers/references/troubleshooting.md +225 -0
- package/.agent/skills/train-sentence-transformers/scripts/mine_hard_negatives.py +199 -0
- package/.agent/skills/train-sentence-transformers/scripts/train_cross_encoder_distillation_example.py +254 -0
- package/.agent/skills/train-sentence-transformers/scripts/train_cross_encoder_example.py +253 -0
- package/.agent/skills/train-sentence-transformers/scripts/train_cross_encoder_listwise_example.py +282 -0
- package/.agent/skills/train-sentence-transformers/scripts/train_multi_vector_encoder_example.py +240 -0
- package/.agent/skills/train-sentence-transformers/scripts/train_sentence_transformer_distillation_example.py +303 -0
- package/.agent/skills/train-sentence-transformers/scripts/train_sentence_transformer_example.py +227 -0
- package/.agent/skills/train-sentence-transformers/scripts/train_sentence_transformer_make_multilingual_example.py +308 -0
- package/.agent/skills/train-sentence-transformers/scripts/train_sentence_transformer_matryoshka_example.py +212 -0
- package/.agent/skills/train-sentence-transformers/scripts/train_sentence_transformer_multi_dataset_example.py +258 -0
- package/.agent/skills/train-sentence-transformers/scripts/train_sentence_transformer_static_embedding_example.py +273 -0
- package/.agent/skills/train-sentence-transformers/scripts/train_sentence_transformer_with_lora_example.py +275 -0
- package/.agent/skills/train-sentence-transformers/scripts/train_sparse_encoder_distillation_example.py +262 -0
- package/.agent/skills/train-sentence-transformers/scripts/train_sparse_encoder_example.py +237 -0
- package/.agent/skills/transformers-js/SKILL.md +692 -0
- package/.agent/skills/transformers-js/references/CACHE.md +339 -0
- package/.agent/skills/transformers-js/references/CONFIGURATION.md +438 -0
- package/.agent/skills/transformers-js/references/EXAMPLES.md +620 -0
- package/.agent/skills/transformers-js/references/MODEL_ARCHITECTURES.md +167 -0
- package/.agent/skills/transformers-js/references/MODEL_REGISTRY.md +197 -0
- package/.agent/skills/transformers-js/references/PIPELINE_OPTIONS.md +557 -0
- package/.agent/skills/transformers-js/references/TEXT_GENERATION.md +315 -0
- package/.agent/skills/tribunal-instincts-memory/SKILL.md +31 -0
- package/.agent/skills/trl-training/SKILL.md +318 -0
- package/.agent/workflows/assimilate.md +21 -0
- package/.agent/workflows/audit.md +5 -0
- package/.agent/workflows/generate.md +5 -0
- package/.agent/workflows/htop.md +22 -0
- package/.agent/workflows/marathon.md +5 -0
- package/.agent/workflows/resume.md +20 -0
- package/.agent/workflows/team-mode.md +22 -0
- package/.agents/skills/agent-syscall-guidelines/SKILL.md +25 -0
- package/.agents/skills/agentic-workflows-2026/SKILL.md +8 -0
- package/.agents/skills/agentshield-security/SKILL.md +26 -0
- package/.agents/skills/context-engineering-pro/SKILL.md +7 -0
- package/.agents/skills/fabel-protocol/SKILL.md +5 -0
- package/.agents/skills/graph-engineering/SKILL.md +28 -0
- package/.agents/skills/harness-protocol/SKILL.md +8 -0
- package/.agents/skills/hf-cli/SKILL.md +239 -0
- package/.agents/skills/hf-cloud-aws-context-discovery/SKILL.md +74 -0
- package/.agents/skills/hf-cloud-python-env-setup/SKILL.md +90 -0
- package/.agents/skills/hf-cloud-python-env-setup/requirements.txt +17 -0
- package/.agents/skills/hf-cloud-python-env-setup/scripts/check_versions.py +29 -0
- package/.agents/skills/hf-cloud-python-env-setup/scripts/setup_env.py +143 -0
- package/.agents/skills/hf-cloud-sagemaker-deployment-planner/SKILL.md +90 -0
- package/.agents/skills/hf-cloud-sagemaker-iam-preflight/SKILL.md +102 -0
- package/.agents/skills/hf-cloud-sagemaker-iam-preflight/references/minimum-permissions.json +47 -0
- package/.agents/skills/hf-cloud-sagemaker-iam-preflight/references/trust-policy.json +12 -0
- package/.agents/skills/hf-cloud-sagemaker-iam-preflight/scripts/check_role.py +184 -0
- package/.agents/skills/hf-cloud-sagemaker-iam-preflight/scripts/create_role.py +118 -0
- package/.agents/skills/hf-cloud-sagemaker-production-defaults/SKILL.md +419 -0
- package/.agents/skills/hf-cloud-sagemaker-production-defaults/references/deployment-template.md +59 -0
- package/.agents/skills/hf-cloud-sagemaker-production-defaults/scripts/_common.py +148 -0
- package/.agents/skills/hf-cloud-sagemaker-production-defaults/scripts/deploy.py +327 -0
- package/.agents/skills/hf-cloud-sagemaker-production-defaults/scripts/deploy_async.py +434 -0
- package/.agents/skills/hf-cloud-sagemaker-production-defaults/scripts/deploy_ic.py +506 -0
- package/.agents/skills/hf-cloud-sagemaker-production-defaults/scripts/invoke_endpoint.py +180 -0
- package/.agents/skills/hf-cloud-sagemaker-production-defaults/scripts/teardown.py +246 -0
- package/.agents/skills/hf-cloud-serving-image-selection/SKILL.md +220 -0
- package/.agents/skills/hf-cloud-serving-image-selection/references/model-to-image.md +111 -0
- package/.agents/skills/hf-cloud-serving-image-selection/scripts/mirror_image.py +145 -0
- package/.agents/skills/hf-mem/SKILL.md +79 -0
- package/.agents/skills/huggingface-best/SKILL.md +134 -0
- package/.agents/skills/huggingface-community-evals/SKILL.md +207 -0
- package/.agents/skills/huggingface-community-evals/examples/.env.example +3 -0
- package/.agents/skills/huggingface-community-evals/examples/USAGE_EXAMPLES.md +101 -0
- package/.agents/skills/huggingface-community-evals/scripts/inspect_eval_uv.py +104 -0
- package/.agents/skills/huggingface-community-evals/scripts/inspect_vllm_uv.py +306 -0
- package/.agents/skills/huggingface-community-evals/scripts/lighteval_vllm_uv.py +297 -0
- package/.agents/skills/huggingface-datasets/SKILL.md +107 -0
- package/.agents/skills/huggingface-gradio/SKILL.md +298 -0
- package/.agents/skills/huggingface-gradio/examples.md +613 -0
- package/.agents/skills/huggingface-llm-trainer/SKILL.md +738 -0
- package/.agents/skills/huggingface-llm-trainer/references/gguf_conversion.md +296 -0
- package/.agents/skills/huggingface-llm-trainer/references/hardware_guide.md +283 -0
- package/.agents/skills/huggingface-llm-trainer/references/hub_saving.md +364 -0
- package/.agents/skills/huggingface-llm-trainer/references/local_training_macos.md +231 -0
- package/.agents/skills/huggingface-llm-trainer/references/reliability_principles.md +371 -0
- package/.agents/skills/huggingface-llm-trainer/references/trackio_guide.md +191 -0
- package/.agents/skills/huggingface-llm-trainer/references/training_methods.md +150 -0
- package/.agents/skills/huggingface-llm-trainer/references/training_patterns.md +203 -0
- package/.agents/skills/huggingface-llm-trainer/references/troubleshooting.md +282 -0
- package/.agents/skills/huggingface-llm-trainer/references/unsloth.md +313 -0
- package/.agents/skills/huggingface-llm-trainer/scripts/convert_to_gguf.py +424 -0
- package/.agents/skills/huggingface-llm-trainer/scripts/dataset_inspector.py +417 -0
- package/.agents/skills/huggingface-llm-trainer/scripts/estimate_cost.py +150 -0
- package/.agents/skills/huggingface-llm-trainer/scripts/hf_benchmarks.py +659 -0
- package/.agents/skills/huggingface-llm-trainer/scripts/train_dpo_example.py +106 -0
- package/.agents/skills/huggingface-llm-trainer/scripts/train_grpo_example.py +89 -0
- package/.agents/skills/huggingface-llm-trainer/scripts/train_sft_example.py +122 -0
- package/.agents/skills/huggingface-llm-trainer/scripts/unsloth_sft_example.py +512 -0
- package/.agents/skills/huggingface-local-models/SKILL.md +113 -0
- package/.agents/skills/huggingface-local-models/references/hardware.md +38 -0
- package/.agents/skills/huggingface-local-models/references/hub-discovery.md +178 -0
- package/.agents/skills/huggingface-local-models/references/quantization.md +256 -0
- package/.agents/skills/huggingface-lora-space-builder/SKILL.md +394 -0
- package/.agents/skills/huggingface-lora-space-builder/references/adapting-to-the-lora.md +113 -0
- package/.agents/skills/huggingface-lora-space-builder/references/base-models/krea-2.md +63 -0
- package/.agents/skills/huggingface-lora-space-builder/references/base-models/ltx.md +205 -0
- package/.agents/skills/huggingface-lora-space-builder/references/base-models/qwen-image.md +144 -0
- package/.agents/skills/huggingface-lora-space-builder/references/creative-mode.md +262 -0
- package/.agents/skills/huggingface-lora-space-builder/references/tasks.md +124 -0
- package/.agents/skills/huggingface-lora-space-builder/references/zerogpu-and-publishing.md +196 -0
- package/.agents/skills/huggingface-paper-publisher/SKILL.md +624 -0
- package/.agents/skills/huggingface-paper-publisher/examples/example_usage.md +326 -0
- package/.agents/skills/huggingface-paper-publisher/references/quick_reference.md +216 -0
- package/.agents/skills/huggingface-paper-publisher/scripts/paper_manager.py +606 -0
- package/.agents/skills/huggingface-paper-publisher/templates/arxiv.md +299 -0
- package/.agents/skills/huggingface-paper-publisher/templates/ml-report.md +358 -0
- package/.agents/skills/huggingface-paper-publisher/templates/modern.md +319 -0
- package/.agents/skills/huggingface-paper-publisher/templates/standard.md +201 -0
- package/.agents/skills/huggingface-papers/SKILL.md +239 -0
- package/.agents/skills/huggingface-spaces/README.md +9 -0
- package/.agents/skills/huggingface-spaces/SKILL.md +242 -0
- package/.agents/skills/huggingface-spaces/references/3d-cuda-extensions.md +109 -0
- package/.agents/skills/huggingface-spaces/references/3d-generation.md +98 -0
- package/.agents/skills/huggingface-spaces/references/3d-gsplat.md +53 -0
- package/.agents/skills/huggingface-spaces/references/3d-models.md +227 -0
- package/.agents/skills/huggingface-spaces/references/3d-outputs.md +114 -0
- package/.agents/skills/huggingface-spaces/references/buckets.md +89 -0
- package/.agents/skills/huggingface-spaces/references/debugging.md +236 -0
- package/.agents/skills/huggingface-spaces/references/gradio.md +200 -0
- package/.agents/skills/huggingface-spaces/references/grants.md +63 -0
- package/.agents/skills/huggingface-spaces/references/inference-providers.md +97 -0
- package/.agents/skills/huggingface-spaces/references/known-errors.md +232 -0
- package/.agents/skills/huggingface-spaces/references/requirements.md +169 -0
- package/.agents/skills/huggingface-spaces/references/zerogpu.md +349 -0
- package/.agents/skills/huggingface-tool-builder/SKILL.md +120 -0
- package/.agents/skills/huggingface-tool-builder/references/baseline_hf_api.py +57 -0
- package/.agents/skills/huggingface-tool-builder/references/baseline_hf_api.sh +40 -0
- package/.agents/skills/huggingface-tool-builder/references/baseline_hf_api.tsx +57 -0
- package/.agents/skills/huggingface-tool-builder/references/find_models_by_paper.sh +230 -0
- package/.agents/skills/huggingface-tool-builder/references/hf_enrich_models.sh +96 -0
- package/.agents/skills/huggingface-tool-builder/references/hf_model_card_frontmatter.sh +188 -0
- package/.agents/skills/huggingface-tool-builder/references/hf_model_papers_auth.sh +171 -0
- package/.agents/skills/huggingface-trackio/SKILL.md +142 -0
- package/.agents/skills/huggingface-trackio/references/alerts.md +196 -0
- package/.agents/skills/huggingface-trackio/references/logging_metrics.md +212 -0
- package/.agents/skills/huggingface-trackio/references/retrieving_metrics.md +251 -0
- package/.agents/skills/huggingface-vision-trainer/SKILL.md +593 -0
- package/.agents/skills/huggingface-vision-trainer/references/finetune_sam2_trainer.md +254 -0
- package/.agents/skills/huggingface-vision-trainer/references/hub_saving.md +618 -0
- package/.agents/skills/huggingface-vision-trainer/references/image_classification_training_notebook.md +279 -0
- package/.agents/skills/huggingface-vision-trainer/references/object_detection_training_notebook.md +700 -0
- package/.agents/skills/huggingface-vision-trainer/references/reliability_principles.md +310 -0
- package/.agents/skills/huggingface-vision-trainer/references/timm_trainer.md +91 -0
- package/.agents/skills/huggingface-vision-trainer/scripts/dataset_inspector.py +814 -0
- package/.agents/skills/huggingface-vision-trainer/scripts/estimate_cost.py +217 -0
- package/.agents/skills/huggingface-vision-trainer/scripts/image_classification_training.py +383 -0
- package/.agents/skills/huggingface-vision-trainer/scripts/object_detection_training.py +710 -0
- package/.agents/skills/huggingface-vision-trainer/scripts/sam_segmentation_training.py +382 -0
- package/.agents/skills/huggingface-zerogpu/SKILL.md +289 -0
- package/.agents/skills/huggingface-zerogpu/references/concurrency.md +79 -0
- package/.agents/skills/huggingface-zerogpu/references/cuda-and-deps.md +66 -0
- package/.agents/skills/huggingface-zerogpu/references/how-quota-works.md +74 -0
- package/.agents/skills/huggingface-zerogpu/references/how-zerogpu-works.md +50 -0
- package/.agents/skills/human-gate-negotiation/SKILL.md +30 -0
- package/.agents/skills/intelligent-routing/SKILL.md +10 -4
- package/.agents/skills/semantic-filesystem-navigation/SKILL.md +24 -0
- package/.agents/skills/session-log-interrogation/SKILL.md +29 -0
- package/.agents/skills/subagent-driven-development/SKILL.md +32 -0
- package/.agents/skills/systematic-debugging/SKILL.md +5 -0
- package/.agents/skills/train-sentence-transformers/SKILL.md +109 -0
- package/.agents/skills/train-sentence-transformers/references/base_model_selection.md +79 -0
- package/.agents/skills/train-sentence-transformers/references/dataset_formats.md +128 -0
- package/.agents/skills/train-sentence-transformers/references/evaluators_cross_encoder.md +116 -0
- package/.agents/skills/train-sentence-transformers/references/evaluators_multi_vector_encoder.md +101 -0
- package/.agents/skills/train-sentence-transformers/references/evaluators_sentence_transformer.md +151 -0
- package/.agents/skills/train-sentence-transformers/references/evaluators_sparse_encoder.md +121 -0
- package/.agents/skills/train-sentence-transformers/references/hardware_guide.md +105 -0
- package/.agents/skills/train-sentence-transformers/references/hf_jobs_execution.md +173 -0
- package/.agents/skills/train-sentence-transformers/references/losses_cross_encoder.md +137 -0
- package/.agents/skills/train-sentence-transformers/references/losses_multi_vector_encoder.md +100 -0
- package/.agents/skills/train-sentence-transformers/references/losses_sentence_transformer.md +246 -0
- package/.agents/skills/train-sentence-transformers/references/losses_sparse_encoder.md +106 -0
- package/.agents/skills/train-sentence-transformers/references/model_architectures.md +178 -0
- package/.agents/skills/train-sentence-transformers/references/prompts_and_instructions.md +160 -0
- package/.agents/skills/train-sentence-transformers/references/training_args.md +270 -0
- package/.agents/skills/train-sentence-transformers/references/troubleshooting.md +225 -0
- package/.agents/skills/train-sentence-transformers/scripts/mine_hard_negatives.py +199 -0
- package/.agents/skills/train-sentence-transformers/scripts/train_cross_encoder_distillation_example.py +254 -0
- package/.agents/skills/train-sentence-transformers/scripts/train_cross_encoder_example.py +253 -0
- package/.agents/skills/train-sentence-transformers/scripts/train_cross_encoder_listwise_example.py +282 -0
- package/.agents/skills/train-sentence-transformers/scripts/train_multi_vector_encoder_example.py +240 -0
- package/.agents/skills/train-sentence-transformers/scripts/train_sentence_transformer_distillation_example.py +303 -0
- package/.agents/skills/train-sentence-transformers/scripts/train_sentence_transformer_example.py +227 -0
- package/.agents/skills/train-sentence-transformers/scripts/train_sentence_transformer_make_multilingual_example.py +308 -0
- package/.agents/skills/train-sentence-transformers/scripts/train_sentence_transformer_matryoshka_example.py +212 -0
- package/.agents/skills/train-sentence-transformers/scripts/train_sentence_transformer_multi_dataset_example.py +258 -0
- package/.agents/skills/train-sentence-transformers/scripts/train_sentence_transformer_static_embedding_example.py +273 -0
- package/.agents/skills/train-sentence-transformers/scripts/train_sentence_transformer_with_lora_example.py +275 -0
- package/.agents/skills/train-sentence-transformers/scripts/train_sparse_encoder_distillation_example.py +262 -0
- package/.agents/skills/train-sentence-transformers/scripts/train_sparse_encoder_example.py +237 -0
- package/.agents/skills/transformers-js/SKILL.md +692 -0
- package/.agents/skills/transformers-js/references/CACHE.md +339 -0
- package/.agents/skills/transformers-js/references/CONFIGURATION.md +438 -0
- package/.agents/skills/transformers-js/references/EXAMPLES.md +620 -0
- package/.agents/skills/transformers-js/references/MODEL_ARCHITECTURES.md +167 -0
- package/.agents/skills/transformers-js/references/MODEL_REGISTRY.md +197 -0
- package/.agents/skills/transformers-js/references/PIPELINE_OPTIONS.md +557 -0
- package/.agents/skills/transformers-js/references/TEXT_GENERATION.md +315 -0
- package/.agents/skills/tribunal-instincts-memory/SKILL.md +31 -0
- package/.agents/skills/trl-training/SKILL.md +318 -0
- package/CONTRIBUTING.md +2 -2
- package/README.md +135 -435
- package/bin/adapter-install.js +1 -1
- package/bin/mcp-server.js +5 -5
- package/dist/tui/banner.js +1 -1
- package/gemini-extension.json +1 -1
- package/mcp_config.json +1 -1
- package/package.json +7 -7
- package/plugin.json +1 -1
- package/scripts/context_compiler.js +1 -1
- package/scripts/skill_enhancement_engine.js +6 -6
- package/skills/agent-syscall-guidelines/SKILL.md +25 -0
- package/skills/agentic-workflows-2026/SKILL.md +8 -0
- package/skills/agentshield-security/SKILL.md +26 -0
- package/skills/context-engineering-pro/SKILL.md +7 -0
- package/skills/fabel-protocol/SKILL.md +5 -0
- package/skills/graph-engineering/SKILL.md +28 -0
- package/skills/harness-protocol/SKILL.md +8 -0
- package/skills/hf-cli/SKILL.md +239 -0
- package/skills/hf-cloud-aws-context-discovery/SKILL.md +74 -0
- package/skills/hf-cloud-python-env-setup/SKILL.md +90 -0
- package/skills/hf-cloud-python-env-setup/requirements.txt +17 -0
- package/skills/hf-cloud-python-env-setup/scripts/check_versions.py +29 -0
- package/skills/hf-cloud-python-env-setup/scripts/setup_env.py +143 -0
- package/skills/hf-cloud-sagemaker-deployment-planner/SKILL.md +90 -0
- package/skills/hf-cloud-sagemaker-iam-preflight/SKILL.md +102 -0
- package/skills/hf-cloud-sagemaker-iam-preflight/references/minimum-permissions.json +47 -0
- package/skills/hf-cloud-sagemaker-iam-preflight/references/trust-policy.json +12 -0
- package/skills/hf-cloud-sagemaker-iam-preflight/scripts/check_role.py +184 -0
- package/skills/hf-cloud-sagemaker-iam-preflight/scripts/create_role.py +118 -0
- package/skills/hf-cloud-sagemaker-production-defaults/SKILL.md +419 -0
- package/skills/hf-cloud-sagemaker-production-defaults/references/deployment-template.md +59 -0
- package/skills/hf-cloud-sagemaker-production-defaults/scripts/_common.py +148 -0
- package/skills/hf-cloud-sagemaker-production-defaults/scripts/deploy.py +327 -0
- package/skills/hf-cloud-sagemaker-production-defaults/scripts/deploy_async.py +434 -0
- package/skills/hf-cloud-sagemaker-production-defaults/scripts/deploy_ic.py +506 -0
- package/skills/hf-cloud-sagemaker-production-defaults/scripts/invoke_endpoint.py +180 -0
- package/skills/hf-cloud-sagemaker-production-defaults/scripts/teardown.py +246 -0
- package/skills/hf-cloud-serving-image-selection/SKILL.md +220 -0
- package/skills/hf-cloud-serving-image-selection/references/model-to-image.md +111 -0
- package/skills/hf-cloud-serving-image-selection/scripts/mirror_image.py +145 -0
- package/skills/hf-mem/SKILL.md +79 -0
- package/skills/huggingface-best/SKILL.md +134 -0
- package/skills/huggingface-community-evals/SKILL.md +207 -0
- package/skills/huggingface-community-evals/examples/.env.example +3 -0
- package/skills/huggingface-community-evals/examples/USAGE_EXAMPLES.md +101 -0
- package/skills/huggingface-community-evals/scripts/inspect_eval_uv.py +104 -0
- package/skills/huggingface-community-evals/scripts/inspect_vllm_uv.py +306 -0
- package/skills/huggingface-community-evals/scripts/lighteval_vllm_uv.py +297 -0
- package/skills/huggingface-datasets/SKILL.md +107 -0
- package/skills/huggingface-gradio/SKILL.md +298 -0
- package/skills/huggingface-gradio/examples.md +613 -0
- package/skills/huggingface-llm-trainer/SKILL.md +738 -0
- package/skills/huggingface-llm-trainer/references/gguf_conversion.md +296 -0
- package/skills/huggingface-llm-trainer/references/hardware_guide.md +283 -0
- package/skills/huggingface-llm-trainer/references/hub_saving.md +364 -0
- package/skills/huggingface-llm-trainer/references/local_training_macos.md +231 -0
- package/skills/huggingface-llm-trainer/references/reliability_principles.md +371 -0
- package/skills/huggingface-llm-trainer/references/trackio_guide.md +191 -0
- package/skills/huggingface-llm-trainer/references/training_methods.md +150 -0
- package/skills/huggingface-llm-trainer/references/training_patterns.md +203 -0
- package/skills/huggingface-llm-trainer/references/troubleshooting.md +282 -0
- package/skills/huggingface-llm-trainer/references/unsloth.md +313 -0
- package/skills/huggingface-llm-trainer/scripts/convert_to_gguf.py +424 -0
- package/skills/huggingface-llm-trainer/scripts/dataset_inspector.py +417 -0
- package/skills/huggingface-llm-trainer/scripts/estimate_cost.py +150 -0
- package/skills/huggingface-llm-trainer/scripts/hf_benchmarks.py +659 -0
- package/skills/huggingface-llm-trainer/scripts/train_dpo_example.py +106 -0
- package/skills/huggingface-llm-trainer/scripts/train_grpo_example.py +89 -0
- package/skills/huggingface-llm-trainer/scripts/train_sft_example.py +122 -0
- package/skills/huggingface-llm-trainer/scripts/unsloth_sft_example.py +512 -0
- package/skills/huggingface-local-models/SKILL.md +113 -0
- package/skills/huggingface-local-models/references/hardware.md +38 -0
- package/skills/huggingface-local-models/references/hub-discovery.md +178 -0
- package/skills/huggingface-local-models/references/quantization.md +256 -0
- package/skills/huggingface-lora-space-builder/SKILL.md +394 -0
- package/skills/huggingface-lora-space-builder/references/adapting-to-the-lora.md +113 -0
- package/skills/huggingface-lora-space-builder/references/base-models/krea-2.md +63 -0
- package/skills/huggingface-lora-space-builder/references/base-models/ltx.md +205 -0
- package/skills/huggingface-lora-space-builder/references/base-models/qwen-image.md +144 -0
- package/skills/huggingface-lora-space-builder/references/creative-mode.md +262 -0
- package/skills/huggingface-lora-space-builder/references/tasks.md +124 -0
- package/skills/huggingface-lora-space-builder/references/zerogpu-and-publishing.md +196 -0
- package/skills/huggingface-paper-publisher/SKILL.md +624 -0
- package/skills/huggingface-paper-publisher/examples/example_usage.md +326 -0
- package/skills/huggingface-paper-publisher/references/quick_reference.md +216 -0
- package/skills/huggingface-paper-publisher/scripts/paper_manager.py +606 -0
- package/skills/huggingface-paper-publisher/templates/arxiv.md +299 -0
- package/skills/huggingface-paper-publisher/templates/ml-report.md +358 -0
- package/skills/huggingface-paper-publisher/templates/modern.md +319 -0
- package/skills/huggingface-paper-publisher/templates/standard.md +201 -0
- package/skills/huggingface-papers/SKILL.md +239 -0
- package/skills/huggingface-spaces/README.md +9 -0
- package/skills/huggingface-spaces/SKILL.md +242 -0
- package/skills/huggingface-spaces/references/3d-cuda-extensions.md +109 -0
- package/skills/huggingface-spaces/references/3d-generation.md +98 -0
- package/skills/huggingface-spaces/references/3d-gsplat.md +53 -0
- package/skills/huggingface-spaces/references/3d-models.md +227 -0
- package/skills/huggingface-spaces/references/3d-outputs.md +114 -0
- package/skills/huggingface-spaces/references/buckets.md +89 -0
- package/skills/huggingface-spaces/references/debugging.md +236 -0
- package/skills/huggingface-spaces/references/gradio.md +200 -0
- package/skills/huggingface-spaces/references/grants.md +63 -0
- package/skills/huggingface-spaces/references/inference-providers.md +97 -0
- package/skills/huggingface-spaces/references/known-errors.md +232 -0
- package/skills/huggingface-spaces/references/requirements.md +169 -0
- package/skills/huggingface-spaces/references/zerogpu.md +349 -0
- package/skills/huggingface-tool-builder/SKILL.md +120 -0
- package/skills/huggingface-tool-builder/references/baseline_hf_api.py +57 -0
- package/skills/huggingface-tool-builder/references/baseline_hf_api.sh +40 -0
- package/skills/huggingface-tool-builder/references/baseline_hf_api.tsx +57 -0
- package/skills/huggingface-tool-builder/references/find_models_by_paper.sh +230 -0
- package/skills/huggingface-tool-builder/references/hf_enrich_models.sh +96 -0
- package/skills/huggingface-tool-builder/references/hf_model_card_frontmatter.sh +188 -0
- package/skills/huggingface-tool-builder/references/hf_model_papers_auth.sh +171 -0
- package/skills/huggingface-trackio/SKILL.md +142 -0
- package/skills/huggingface-trackio/references/alerts.md +196 -0
- package/skills/huggingface-trackio/references/logging_metrics.md +212 -0
- package/skills/huggingface-trackio/references/retrieving_metrics.md +251 -0
- package/skills/huggingface-vision-trainer/SKILL.md +593 -0
- package/skills/huggingface-vision-trainer/references/finetune_sam2_trainer.md +254 -0
- package/skills/huggingface-vision-trainer/references/hub_saving.md +618 -0
- package/skills/huggingface-vision-trainer/references/image_classification_training_notebook.md +279 -0
- package/skills/huggingface-vision-trainer/references/object_detection_training_notebook.md +700 -0
- package/skills/huggingface-vision-trainer/references/reliability_principles.md +310 -0
- package/skills/huggingface-vision-trainer/references/timm_trainer.md +91 -0
- package/skills/huggingface-vision-trainer/scripts/dataset_inspector.py +814 -0
- package/skills/huggingface-vision-trainer/scripts/estimate_cost.py +217 -0
- package/skills/huggingface-vision-trainer/scripts/image_classification_training.py +383 -0
- package/skills/huggingface-vision-trainer/scripts/object_detection_training.py +710 -0
- package/skills/huggingface-vision-trainer/scripts/sam_segmentation_training.py +382 -0
- package/skills/huggingface-zerogpu/SKILL.md +289 -0
- package/skills/huggingface-zerogpu/references/concurrency.md +79 -0
- package/skills/huggingface-zerogpu/references/cuda-and-deps.md +66 -0
- package/skills/huggingface-zerogpu/references/how-quota-works.md +74 -0
- package/skills/huggingface-zerogpu/references/how-zerogpu-works.md +50 -0
- package/skills/human-gate-negotiation/SKILL.md +30 -0
- package/skills/intelligent-routing/SKILL.md +10 -4
- package/skills/semantic-filesystem-navigation/SKILL.md +24 -0
- package/skills/session-log-interrogation/SKILL.md +29 -0
- package/skills/subagent-driven-development/SKILL.md +32 -0
- package/skills/systematic-debugging/SKILL.md +5 -0
- package/skills/train-sentence-transformers/SKILL.md +109 -0
- package/skills/train-sentence-transformers/references/base_model_selection.md +79 -0
- package/skills/train-sentence-transformers/references/dataset_formats.md +128 -0
- package/skills/train-sentence-transformers/references/evaluators_cross_encoder.md +116 -0
- package/skills/train-sentence-transformers/references/evaluators_multi_vector_encoder.md +101 -0
- package/skills/train-sentence-transformers/references/evaluators_sentence_transformer.md +151 -0
- package/skills/train-sentence-transformers/references/evaluators_sparse_encoder.md +121 -0
- package/skills/train-sentence-transformers/references/hardware_guide.md +105 -0
- package/skills/train-sentence-transformers/references/hf_jobs_execution.md +173 -0
- package/skills/train-sentence-transformers/references/losses_cross_encoder.md +137 -0
- package/skills/train-sentence-transformers/references/losses_multi_vector_encoder.md +100 -0
- package/skills/train-sentence-transformers/references/losses_sentence_transformer.md +246 -0
- package/skills/train-sentence-transformers/references/losses_sparse_encoder.md +106 -0
- package/skills/train-sentence-transformers/references/model_architectures.md +178 -0
- package/skills/train-sentence-transformers/references/prompts_and_instructions.md +160 -0
- package/skills/train-sentence-transformers/references/training_args.md +270 -0
- package/skills/train-sentence-transformers/references/troubleshooting.md +225 -0
- package/skills/train-sentence-transformers/scripts/mine_hard_negatives.py +199 -0
- package/skills/train-sentence-transformers/scripts/train_cross_encoder_distillation_example.py +254 -0
- package/skills/train-sentence-transformers/scripts/train_cross_encoder_example.py +253 -0
- package/skills/train-sentence-transformers/scripts/train_cross_encoder_listwise_example.py +282 -0
- package/skills/train-sentence-transformers/scripts/train_multi_vector_encoder_example.py +240 -0
- package/skills/train-sentence-transformers/scripts/train_sentence_transformer_distillation_example.py +303 -0
- package/skills/train-sentence-transformers/scripts/train_sentence_transformer_example.py +227 -0
- package/skills/train-sentence-transformers/scripts/train_sentence_transformer_make_multilingual_example.py +308 -0
- package/skills/train-sentence-transformers/scripts/train_sentence_transformer_matryoshka_example.py +212 -0
- package/skills/train-sentence-transformers/scripts/train_sentence_transformer_multi_dataset_example.py +258 -0
- package/skills/train-sentence-transformers/scripts/train_sentence_transformer_static_embedding_example.py +273 -0
- package/skills/train-sentence-transformers/scripts/train_sentence_transformer_with_lora_example.py +275 -0
- package/skills/train-sentence-transformers/scripts/train_sparse_encoder_distillation_example.py +262 -0
- package/skills/train-sentence-transformers/scripts/train_sparse_encoder_example.py +237 -0
- package/skills/transformers-js/SKILL.md +692 -0
- package/skills/transformers-js/references/CACHE.md +339 -0
- package/skills/transformers-js/references/CONFIGURATION.md +438 -0
- package/skills/transformers-js/references/EXAMPLES.md +620 -0
- package/skills/transformers-js/references/MODEL_ARCHITECTURES.md +167 -0
- package/skills/transformers-js/references/MODEL_REGISTRY.md +197 -0
- package/skills/transformers-js/references/PIPELINE_OPTIONS.md +557 -0
- package/skills/transformers-js/references/TEXT_GENERATION.md +315 -0
- package/skills/tribunal-instincts-memory/SKILL.md +31 -0
- package/skills/trl-training/SKILL.md +318 -0
|
@@ -0,0 +1,315 @@
|
|
|
1
|
+
# Text Generation Guide
|
|
2
|
+
|
|
3
|
+
Guide to generating text with Transformers.js, including streaming and chat format.
|
|
4
|
+
|
|
5
|
+
## Table of Contents
|
|
6
|
+
|
|
7
|
+
1. [Basic Generation](#basic-generation)
|
|
8
|
+
2. [Streaming](#streaming)
|
|
9
|
+
3. [Chat Format](#chat-format)
|
|
10
|
+
4. [Generation Parameters](#generation-parameters)
|
|
11
|
+
5. [Model Selection](#model-selection)
|
|
12
|
+
6. [Best Practices](#best-practices)
|
|
13
|
+
|
|
14
|
+
## Basic Generation
|
|
15
|
+
|
|
16
|
+
```javascript
|
|
17
|
+
import { pipeline } from '@huggingface/transformers';
|
|
18
|
+
|
|
19
|
+
const generator = await pipeline(
|
|
20
|
+
'text-generation',
|
|
21
|
+
'onnx-community/Qwen2.5-0.5B-Instruct',
|
|
22
|
+
{ dtype: 'q4' }
|
|
23
|
+
);
|
|
24
|
+
|
|
25
|
+
const result = await generator('Once upon a time', {
|
|
26
|
+
max_new_tokens: 100,
|
|
27
|
+
temperature: 0.7,
|
|
28
|
+
});
|
|
29
|
+
|
|
30
|
+
console.log(result[0].generated_text);
|
|
31
|
+
|
|
32
|
+
// Clean up when done
|
|
33
|
+
await generator.dispose();
|
|
34
|
+
```
|
|
35
|
+
|
|
36
|
+
## Streaming
|
|
37
|
+
|
|
38
|
+
Stream tokens as they're generated for better UX. Once you understand streaming, you can combine it with other features like chat format.
|
|
39
|
+
|
|
40
|
+
### Node.js
|
|
41
|
+
|
|
42
|
+
```javascript
|
|
43
|
+
import { pipeline, TextStreamer } from '@huggingface/transformers';
|
|
44
|
+
|
|
45
|
+
const generator = await pipeline(
|
|
46
|
+
'text-generation',
|
|
47
|
+
'onnx-community/Qwen2.5-0.5B-Instruct',
|
|
48
|
+
{ dtype: 'q4' }
|
|
49
|
+
);
|
|
50
|
+
|
|
51
|
+
const streamer = new TextStreamer(generator.tokenizer, {
|
|
52
|
+
skip_prompt: true,
|
|
53
|
+
skip_special_tokens: true,
|
|
54
|
+
callback_function: (token) => {
|
|
55
|
+
process.stdout.write(token);
|
|
56
|
+
},
|
|
57
|
+
});
|
|
58
|
+
|
|
59
|
+
await generator('Tell me a story', {
|
|
60
|
+
max_new_tokens: 200,
|
|
61
|
+
temperature: 0.7,
|
|
62
|
+
streamer,
|
|
63
|
+
});
|
|
64
|
+
```
|
|
65
|
+
|
|
66
|
+
### Browser
|
|
67
|
+
|
|
68
|
+
```html
|
|
69
|
+
<!DOCTYPE html>
|
|
70
|
+
<html>
|
|
71
|
+
<body>
|
|
72
|
+
<textarea id="prompt" placeholder="Enter prompt..."></textarea>
|
|
73
|
+
<button onclick="generate()">Generate</button>
|
|
74
|
+
<div id="output"></div>
|
|
75
|
+
|
|
76
|
+
<script type="module">
|
|
77
|
+
import { pipeline, TextStreamer } from 'https://cdn.jsdelivr.net/npm/@huggingface/transformers@4';
|
|
78
|
+
|
|
79
|
+
const generator = await pipeline(
|
|
80
|
+
'text-generation',
|
|
81
|
+
'onnx-community/Qwen2.5-0.5B-Instruct',
|
|
82
|
+
{ dtype: 'q4' }
|
|
83
|
+
);
|
|
84
|
+
|
|
85
|
+
window.generate = async function() {
|
|
86
|
+
const prompt = document.getElementById('prompt').value;
|
|
87
|
+
const outputDiv = document.getElementById('output');
|
|
88
|
+
outputDiv.textContent = '';
|
|
89
|
+
|
|
90
|
+
const streamer = new TextStreamer(generator.tokenizer, {
|
|
91
|
+
skip_prompt: true,
|
|
92
|
+
skip_special_tokens: true,
|
|
93
|
+
callback_function: (token) => {
|
|
94
|
+
outputDiv.textContent += token;
|
|
95
|
+
},
|
|
96
|
+
});
|
|
97
|
+
|
|
98
|
+
await generator(prompt, {
|
|
99
|
+
max_new_tokens: 200,
|
|
100
|
+
temperature: 0.7,
|
|
101
|
+
streamer,
|
|
102
|
+
});
|
|
103
|
+
};
|
|
104
|
+
</script>
|
|
105
|
+
</body>
|
|
106
|
+
</html>
|
|
107
|
+
```
|
|
108
|
+
|
|
109
|
+
### React
|
|
110
|
+
|
|
111
|
+
```jsx
|
|
112
|
+
import { useState, useRef, useEffect } from 'react';
|
|
113
|
+
import { pipeline, TextStreamer } from '@huggingface/transformers';
|
|
114
|
+
|
|
115
|
+
function StreamingGenerator() {
|
|
116
|
+
const generatorRef = useRef(null);
|
|
117
|
+
const [output, setOutput] = useState('');
|
|
118
|
+
const [loading, setLoading] = useState(false);
|
|
119
|
+
|
|
120
|
+
const handleGenerate = async (prompt) => {
|
|
121
|
+
if (!prompt) return;
|
|
122
|
+
|
|
123
|
+
setLoading(true);
|
|
124
|
+
setOutput('');
|
|
125
|
+
|
|
126
|
+
// Load model on first generate
|
|
127
|
+
if (!generatorRef.current) {
|
|
128
|
+
generatorRef.current = await pipeline(
|
|
129
|
+
'text-generation',
|
|
130
|
+
'onnx-community/Qwen2.5-0.5B-Instruct',
|
|
131
|
+
{ dtype: 'q4' }
|
|
132
|
+
);
|
|
133
|
+
}
|
|
134
|
+
|
|
135
|
+
const streamer = new TextStreamer(generatorRef.current.tokenizer, {
|
|
136
|
+
skip_prompt: true,
|
|
137
|
+
skip_special_tokens: true,
|
|
138
|
+
callback_function: (token) => {
|
|
139
|
+
setOutput((prev) => prev + token);
|
|
140
|
+
},
|
|
141
|
+
});
|
|
142
|
+
|
|
143
|
+
await generatorRef.current(prompt, {
|
|
144
|
+
max_new_tokens: 200,
|
|
145
|
+
temperature: 0.7,
|
|
146
|
+
streamer,
|
|
147
|
+
});
|
|
148
|
+
|
|
149
|
+
setLoading(false);
|
|
150
|
+
};
|
|
151
|
+
|
|
152
|
+
// Cleanup on unmount
|
|
153
|
+
useEffect(() => {
|
|
154
|
+
return () => {
|
|
155
|
+
if (generatorRef.current) {
|
|
156
|
+
generatorRef.current.dispose();
|
|
157
|
+
}
|
|
158
|
+
};
|
|
159
|
+
}, []);
|
|
160
|
+
|
|
161
|
+
return (
|
|
162
|
+
<div>
|
|
163
|
+
<button onClick={() => handleGenerate('Tell me a story')} disabled={loading}>
|
|
164
|
+
{loading ? 'Generating...' : 'Generate'}
|
|
165
|
+
</button>
|
|
166
|
+
<div>{output}</div>
|
|
167
|
+
</div>
|
|
168
|
+
);
|
|
169
|
+
}
|
|
170
|
+
```
|
|
171
|
+
|
|
172
|
+
## Chat Format
|
|
173
|
+
|
|
174
|
+
Use structured messages for conversations. Works with both basic generation and streaming (just add `streamer` parameter).
|
|
175
|
+
|
|
176
|
+
### Single Turn
|
|
177
|
+
|
|
178
|
+
```javascript
|
|
179
|
+
import { pipeline } from '@huggingface/transformers';
|
|
180
|
+
|
|
181
|
+
const generator = await pipeline(
|
|
182
|
+
'text-generation',
|
|
183
|
+
'onnx-community/Qwen2.5-0.5B-Instruct',
|
|
184
|
+
{ dtype: 'q4' }
|
|
185
|
+
);
|
|
186
|
+
|
|
187
|
+
const messages = [
|
|
188
|
+
{ role: 'system', content: 'You are a helpful assistant.' },
|
|
189
|
+
{ role: 'user', content: 'How do I create an async function?' }
|
|
190
|
+
];
|
|
191
|
+
|
|
192
|
+
const result = await generator(messages, {
|
|
193
|
+
max_new_tokens: 256,
|
|
194
|
+
temperature: 0.7,
|
|
195
|
+
});
|
|
196
|
+
|
|
197
|
+
console.log(result[0].generated_text);
|
|
198
|
+
```
|
|
199
|
+
|
|
200
|
+
### Multi-turn Conversation
|
|
201
|
+
|
|
202
|
+
```javascript
|
|
203
|
+
const conversation = [
|
|
204
|
+
{ role: 'system', content: 'You are a helpful assistant.' },
|
|
205
|
+
{ role: 'user', content: 'What is JavaScript?' },
|
|
206
|
+
{ role: 'assistant', content: 'JavaScript is a programming language...' },
|
|
207
|
+
{ role: 'user', content: 'Can you show an example?' }
|
|
208
|
+
];
|
|
209
|
+
|
|
210
|
+
const result = await generator(conversation, {
|
|
211
|
+
max_new_tokens: 200,
|
|
212
|
+
temperature: 0.7,
|
|
213
|
+
});
|
|
214
|
+
|
|
215
|
+
// To add streaming, just pass a streamer:
|
|
216
|
+
// streamer: new TextStreamer(generator.tokenizer, {...})
|
|
217
|
+
```
|
|
218
|
+
|
|
219
|
+
## Generation Parameters
|
|
220
|
+
|
|
221
|
+
### Common Parameters
|
|
222
|
+
|
|
223
|
+
```javascript
|
|
224
|
+
await generator(prompt, {
|
|
225
|
+
// Token limits
|
|
226
|
+
max_new_tokens: 512, // Maximum tokens to generate
|
|
227
|
+
min_new_tokens: 0, // Minimum tokens to generate
|
|
228
|
+
|
|
229
|
+
// Sampling
|
|
230
|
+
temperature: 0.7, // Randomness (0.0-2.0)
|
|
231
|
+
top_k: 50, // Consider top K tokens
|
|
232
|
+
top_p: 0.95, // Nucleus sampling
|
|
233
|
+
do_sample: true, // Use random sampling (false = always pick most likely token)
|
|
234
|
+
|
|
235
|
+
// Repetition control
|
|
236
|
+
repetition_penalty: 1.0, // Penalty for repeating (1.0 = no penalty)
|
|
237
|
+
no_repeat_ngram_size: 0, // Prevent repeating n-grams
|
|
238
|
+
|
|
239
|
+
// Streaming
|
|
240
|
+
streamer: streamer, // TextStreamer instance
|
|
241
|
+
});
|
|
242
|
+
```
|
|
243
|
+
|
|
244
|
+
### Parameter Effects
|
|
245
|
+
|
|
246
|
+
**Temperature:**
|
|
247
|
+
- Low (0.1-0.5): More focused and deterministic
|
|
248
|
+
- Medium (0.6-0.9): Balanced creativity and coherence
|
|
249
|
+
- High (1.0-2.0): More creative and random
|
|
250
|
+
|
|
251
|
+
```javascript
|
|
252
|
+
// Focused output
|
|
253
|
+
await generator(prompt, { temperature: 0.3, max_new_tokens: 100 });
|
|
254
|
+
|
|
255
|
+
// Creative output
|
|
256
|
+
await generator(prompt, { temperature: 1.2, max_new_tokens: 100 });
|
|
257
|
+
```
|
|
258
|
+
|
|
259
|
+
**Sampling Methods:**
|
|
260
|
+
|
|
261
|
+
```javascript
|
|
262
|
+
// Greedy (deterministic)
|
|
263
|
+
await generator(prompt, {
|
|
264
|
+
do_sample: false,
|
|
265
|
+
max_new_tokens: 100
|
|
266
|
+
});
|
|
267
|
+
|
|
268
|
+
// Top-k sampling
|
|
269
|
+
await generator(prompt, {
|
|
270
|
+
top_k: 50,
|
|
271
|
+
temperature: 0.7,
|
|
272
|
+
max_new_tokens: 100
|
|
273
|
+
});
|
|
274
|
+
|
|
275
|
+
// Top-p (nucleus) sampling
|
|
276
|
+
await generator(prompt, {
|
|
277
|
+
top_p: 0.95,
|
|
278
|
+
temperature: 0.7,
|
|
279
|
+
max_new_tokens: 100
|
|
280
|
+
});
|
|
281
|
+
```
|
|
282
|
+
|
|
283
|
+
## Model Selection
|
|
284
|
+
|
|
285
|
+
Browse available text generation models on Hugging Face Hub:
|
|
286
|
+
|
|
287
|
+
**https://huggingface.co/models?pipeline_tag=text-generation&library=transformers.js&sort=trending**
|
|
288
|
+
|
|
289
|
+
### Selection Tips
|
|
290
|
+
|
|
291
|
+
- **Small models (< 1B params)**: Fast, browser-friendly, use `dtype: 'q4'`
|
|
292
|
+
- **Medium models (1-3B params)**: Balanced quality/speed, use `dtype: 'q4'` or `fp16`
|
|
293
|
+
- **Large models (> 3B params)**: High quality, slower, best for Node.js with `dtype: 'fp16'`
|
|
294
|
+
|
|
295
|
+
Check model cards for:
|
|
296
|
+
- Parameter count and model size
|
|
297
|
+
- Supported languages
|
|
298
|
+
- Benchmark scores
|
|
299
|
+
- License restrictions
|
|
300
|
+
|
|
301
|
+
## Best Practices
|
|
302
|
+
|
|
303
|
+
1. **Model Size**: Use quantized models (`q4`) for browsers, larger models (`fp16`) for servers
|
|
304
|
+
2. **Streaming**: Use streaming for better UX - shows progress and feels responsive
|
|
305
|
+
3. **Token Limits**: Set `max_new_tokens` to prevent runaway generation
|
|
306
|
+
4. **Temperature**: Tune based on use case (creative: 0.8-1.2, factual: 0.3-0.7)
|
|
307
|
+
5. **Memory**: Always call `dispose()` when done
|
|
308
|
+
6. **Caching**: Load model once, reuse for multiple requests
|
|
309
|
+
|
|
310
|
+
## Related Documentation
|
|
311
|
+
|
|
312
|
+
- [Pipeline Options](./PIPELINE_OPTIONS.md) - Configure pipeline loading
|
|
313
|
+
- [Configuration Reference](./CONFIGURATION.md) - Environment settings
|
|
314
|
+
- [Code Examples](./EXAMPLES.md) - More examples for different runtimes
|
|
315
|
+
- [Main Skill Guide](../SKILL.md) - Getting started guide
|
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
---
|
|
2
|
+
name: tribunal-instincts-memory
|
|
3
|
+
description: "Use when extracting lessons from session logs, resolving repeated failures, creating persistent context, and validating AI memory constraints."
|
|
4
|
+
version: 1.0.0
|
|
5
|
+
last-updated: 2026-09-25
|
|
6
|
+
skills:
|
|
7
|
+
- fabel-protocol
|
|
8
|
+
tools: Read, Grep, Glob, Bash, Edit, Write
|
|
9
|
+
scripts-binding:
|
|
10
|
+
- .agent/scripts/memory_archivist.js
|
|
11
|
+
---
|
|
12
|
+
|
|
13
|
+
# Tribunal Instincts (Continuous Memory)
|
|
14
|
+
|
|
15
|
+
## Overview
|
|
16
|
+
Tribunal Instincts allows the agent harness to learn from its past mistakes and successful resolutions without blowing up the context window. It extracts concrete lessons from the Event Log and injects them selectively into future sessions.
|
|
17
|
+
|
|
18
|
+
## Memory Lifecycle
|
|
19
|
+
1. **Candidate Extraction**: The `memory-archivist` scans the durable JSONL session log for anomalies, repeated errors, or successful recoveries.
|
|
20
|
+
2. **Evidence Validation**: A memory CANNOT be created without a specific Event ID or stack trace proving it happened. Subjective complaints are dropped.
|
|
21
|
+
3. **Storage**: Approved lessons are saved to `instincts.json` under specific scopes (e.g., `database`, `ui`).
|
|
22
|
+
4. **Context Injection**: During workflow initialization, `context_broker.js` reads only the instincts relevant to the current task.
|
|
23
|
+
|
|
24
|
+
## Rules of Instinct Retrieval
|
|
25
|
+
- **Token Budget**: Never inject more than 5 instincts at a time.
|
|
26
|
+
- **Conflict Resolution**: If a newer instinct contradicts an older one, the older one is archived.
|
|
27
|
+
- **System Rule Supremacy**: Instincts can NEVER override hardcoded system rules (e.g., `GEMINI.md`). They serve only as technical implementation hints.
|
|
28
|
+
|
|
29
|
+
## Hallucination Traps
|
|
30
|
+
- ❌ Trusting a memory that says "Always use library X" without checking `package.json`.
|
|
31
|
+
- ❌ Creating an instinct for a typo. (Only systemic or architectural lessons deserve permanent memory).
|
|
@@ -0,0 +1,318 @@
|
|
|
1
|
+
---
|
|
2
|
+
name: trl-training
|
|
3
|
+
description: Train and fine-tune transformer language models using TRL (Transformers Reinforcement Learning). Supports SFT, DPO, GRPO, KTO, RLOO and Reward Model training via CLI commands.
|
|
4
|
+
license: Apache-2.0
|
|
5
|
+
metadata:
|
|
6
|
+
version: "1.0.0"
|
|
7
|
+
author: huggingface
|
|
8
|
+
commands: trl sft, trl dpo, trl grpo, trl kto, trl rloo, trl reward
|
|
9
|
+
categories: machine-learning, llm-training, reinforcement-learning
|
|
10
|
+
tags: rlhf, supervised-fine-tuning, dpo, grpo, huggingface, transformers
|
|
11
|
+
documentation: https://huggingface.co/docs/trl/en/clis
|
|
12
|
+
---
|
|
13
|
+
|
|
14
|
+
# TRL Training Skill
|
|
15
|
+
|
|
16
|
+
You are an expert at using the TRL (Transformers Reinforcement Learning) library to train and fine-tune large language models.
|
|
17
|
+
|
|
18
|
+
## Overview
|
|
19
|
+
|
|
20
|
+
TRL provides CLI commands for post-training foundation models using state-of-the-art techniques:
|
|
21
|
+
|
|
22
|
+
- **SFT** (Supervised Fine-Tuning): Fine-tune models on instruction-following or conversational datasets
|
|
23
|
+
- **DPO** (Direct Preference Optimization): Align models using preference data
|
|
24
|
+
- **GRPO** (Group Relative Policy Optimization): Train models by ranking multiple sampled outputs relative to each other and optimizing based on their comparative rewards.
|
|
25
|
+
- **RLOO** (Reinforce Leave One Out): Online RL training with generation-based rewards
|
|
26
|
+
- **Reward Model Training**: Train reward models for RLHF
|
|
27
|
+
|
|
28
|
+
TRL is built on top of Hugging Face Transformers and Accelerate, providing seamless integration with the Hugging Face ecosystem.
|
|
29
|
+
|
|
30
|
+
## Core Commands
|
|
31
|
+
|
|
32
|
+
### trl sft - Supervised Fine-Tuning
|
|
33
|
+
|
|
34
|
+
Fine-tune language models on instruction-following or conversational datasets.
|
|
35
|
+
|
|
36
|
+
**Full training:**
|
|
37
|
+
|
|
38
|
+
```bash
|
|
39
|
+
trl sft \
|
|
40
|
+
--model_name_or_path Qwen/Qwen2-0.5B \
|
|
41
|
+
--dataset_name trl-lib/Capybara \
|
|
42
|
+
--learning_rate 2.0e-5 \
|
|
43
|
+
--num_train_epochs 1 \
|
|
44
|
+
--packing \
|
|
45
|
+
--per_device_train_batch_size 2 \
|
|
46
|
+
--gradient_accumulation_steps 8 \
|
|
47
|
+
--eos_token '<|im_end|>' \
|
|
48
|
+
--eval_strategy steps \
|
|
49
|
+
--eval_steps 100 \
|
|
50
|
+
--output_dir Qwen2-0.5B-SFT \
|
|
51
|
+
--push_to_hub
|
|
52
|
+
```
|
|
53
|
+
|
|
54
|
+
**Train with LoRA adapters:**
|
|
55
|
+
|
|
56
|
+
```bash
|
|
57
|
+
trl sft \
|
|
58
|
+
--model_name_or_path Qwen/Qwen2-0.5B \
|
|
59
|
+
--dataset_name trl-lib/Capybara \
|
|
60
|
+
--learning_rate 2.0e-4 \
|
|
61
|
+
--num_train_epochs 1 \
|
|
62
|
+
--packing \
|
|
63
|
+
--per_device_train_batch_size 2 \
|
|
64
|
+
--gradient_accumulation_steps 8 \
|
|
65
|
+
--eos_token '<|im_end|>' \
|
|
66
|
+
--eval_strategy steps \
|
|
67
|
+
--eval_steps 100 \
|
|
68
|
+
--use_peft \
|
|
69
|
+
--lora_r 32 \
|
|
70
|
+
--lora_alpha 16 \
|
|
71
|
+
--output_dir Qwen2-0.5B-SFT \
|
|
72
|
+
--push_to_hub
|
|
73
|
+
```
|
|
74
|
+
|
|
75
|
+
### trl dpo - Direct Preference Optimization
|
|
76
|
+
|
|
77
|
+
Align models using preference data (chosen/rejected pairs).
|
|
78
|
+
|
|
79
|
+
**Full training:**
|
|
80
|
+
|
|
81
|
+
```bash
|
|
82
|
+
trl dpo \
|
|
83
|
+
--dataset_name trl-lib/ultrafeedback_binarized \
|
|
84
|
+
--model_name_or_path Qwen/Qwen2-0.5B-Instruct \
|
|
85
|
+
--learning_rate 5.0e-7 \
|
|
86
|
+
--num_train_epochs 1 \
|
|
87
|
+
--per_device_train_batch_size 2 \
|
|
88
|
+
--max_steps 1000 \
|
|
89
|
+
--gradient_accumulation_steps 8 \
|
|
90
|
+
--eval_strategy steps \
|
|
91
|
+
--eval_steps 50 \
|
|
92
|
+
--output_dir Qwen2-0.5B-DPO \
|
|
93
|
+
--no_remove_unused_columns
|
|
94
|
+
```
|
|
95
|
+
|
|
96
|
+
**Train with LoRA adapters:**
|
|
97
|
+
|
|
98
|
+
```bash
|
|
99
|
+
trl dpo \
|
|
100
|
+
--dataset_name trl-lib/ultrafeedback_binarized \
|
|
101
|
+
--model_name_or_path Qwen/Qwen2-0.5B-Instruct \
|
|
102
|
+
--learning_rate 5.0e-6 \
|
|
103
|
+
--num_train_epochs 1 \
|
|
104
|
+
--per_device_train_batch_size 2 \
|
|
105
|
+
--max_steps 1000 \
|
|
106
|
+
--gradient_accumulation_steps 8 \
|
|
107
|
+
--eval_strategy steps \
|
|
108
|
+
--eval_steps 50 \
|
|
109
|
+
--output_dir Qwen2-0.5B-DPO \
|
|
110
|
+
--no_remove_unused_columns \
|
|
111
|
+
--use_peft \
|
|
112
|
+
--lora_r 32 \
|
|
113
|
+
--lora_alpha 16
|
|
114
|
+
```
|
|
115
|
+
|
|
116
|
+
### trl grpo - Group Relative Policy Optimization
|
|
117
|
+
|
|
118
|
+
Train models using reward functions or LLM-as-a-judge for evaluating generations and providing rewards.
|
|
119
|
+
|
|
120
|
+
**Basic usage:**
|
|
121
|
+
|
|
122
|
+
```bash
|
|
123
|
+
trl grpo \
|
|
124
|
+
--model_name_or_path Qwen/Qwen2.5-0.5B \
|
|
125
|
+
--dataset_name trl-lib/gsm8k \
|
|
126
|
+
--reward_funcs accuracy_reward \
|
|
127
|
+
--output_dir Qwen2-0.5B-GRPO \
|
|
128
|
+
--push_to_hub
|
|
129
|
+
```
|
|
130
|
+
|
|
131
|
+
### trl rloo - Reinforce Leave One Out
|
|
132
|
+
|
|
133
|
+
Online RL training where the model generates text and receives rewards based on custom criteria.
|
|
134
|
+
|
|
135
|
+
**Basic usage:**
|
|
136
|
+
|
|
137
|
+
```bash
|
|
138
|
+
trl rloo \
|
|
139
|
+
--model_name_or_path Qwen/Qwen2.5-0.5B \
|
|
140
|
+
--dataset_name trl-lib/tldr \
|
|
141
|
+
--reward_model_name_or_path sentiment-analysis:nlptown/bert-base-multilingual-uncased-sentiment \
|
|
142
|
+
--output_dir Qwen2-0.5B-RLOO \
|
|
143
|
+
--push_to_hub
|
|
144
|
+
```
|
|
145
|
+
|
|
146
|
+
### trl reward - Reward Model Training
|
|
147
|
+
|
|
148
|
+
Train a reward model to score text quality for RLHF.
|
|
149
|
+
|
|
150
|
+
**Full training:**
|
|
151
|
+
|
|
152
|
+
```bash
|
|
153
|
+
trl reward \
|
|
154
|
+
--model_name_or_path Qwen/Qwen2-0.5B-Instruct \
|
|
155
|
+
--dataset_name trl-lib/ultrafeedback_binarized \
|
|
156
|
+
--output_dir Qwen2-0.5B-Reward \
|
|
157
|
+
--per_device_train_batch_size 8 \
|
|
158
|
+
--num_train_epochs 1 \
|
|
159
|
+
--learning_rate 1.0e-5 \
|
|
160
|
+
--eval_strategy steps \
|
|
161
|
+
--eval_steps 50 \
|
|
162
|
+
--max_length 2048
|
|
163
|
+
```
|
|
164
|
+
|
|
165
|
+
**Train with LoRA adapters:**
|
|
166
|
+
|
|
167
|
+
```bash
|
|
168
|
+
trl reward \
|
|
169
|
+
--model_name_or_path Qwen/Qwen2-0.5B-Instruct \
|
|
170
|
+
--dataset_name trl-lib/ultrafeedback_binarized \
|
|
171
|
+
--output_dir Qwen2-0.5B-Reward-LoRA \
|
|
172
|
+
--per_device_train_batch_size 8 \
|
|
173
|
+
--num_train_epochs 1 \
|
|
174
|
+
--learning_rate 1.0e-4 \
|
|
175
|
+
--eval_strategy steps \
|
|
176
|
+
--eval_steps 50 \
|
|
177
|
+
--max_length 2048 \
|
|
178
|
+
--use_peft \
|
|
179
|
+
--lora_task_type SEQ_CLS \
|
|
180
|
+
--lora_r 32 \
|
|
181
|
+
--lora_alpha 16
|
|
182
|
+
```
|
|
183
|
+
|
|
184
|
+
## Configuration Files
|
|
185
|
+
|
|
186
|
+
TRL supports YAML configuration files for reproducible training. All CLI arguments can be specified in a config file.
|
|
187
|
+
|
|
188
|
+
**Example config (sft_config.yaml):**
|
|
189
|
+
|
|
190
|
+
```yaml
|
|
191
|
+
model_name_or_path: Qwen/Qwen2.5-0.5B
|
|
192
|
+
dataset_name: trl-lib/Capybara
|
|
193
|
+
learning_rate: 2.0e-5
|
|
194
|
+
num_train_epochs: 1
|
|
195
|
+
per_device_train_batch_size: 8
|
|
196
|
+
gradient_accumulation_steps: 2
|
|
197
|
+
output_dir: ./sft_output
|
|
198
|
+
use_peft: true
|
|
199
|
+
lora_r: 16
|
|
200
|
+
lora_alpha: 16
|
|
201
|
+
report_to: trackio
|
|
202
|
+
```
|
|
203
|
+
|
|
204
|
+
**Launch with config:**
|
|
205
|
+
|
|
206
|
+
```bash
|
|
207
|
+
trl sft --config sft_config.yaml
|
|
208
|
+
```
|
|
209
|
+
|
|
210
|
+
**Override config values:**
|
|
211
|
+
|
|
212
|
+
```bash
|
|
213
|
+
trl sft --config sft_config.yaml --learning_rate 1.0e-5
|
|
214
|
+
```
|
|
215
|
+
|
|
216
|
+
## Distributed Training
|
|
217
|
+
|
|
218
|
+
TRL integrates with Accelerate for multi-GPU and multi-node training.
|
|
219
|
+
|
|
220
|
+
**Multi-GPU training:**
|
|
221
|
+
|
|
222
|
+
```bash
|
|
223
|
+
trl sft \
|
|
224
|
+
--config sft_config.yaml \
|
|
225
|
+
--num_processes 4
|
|
226
|
+
```
|
|
227
|
+
|
|
228
|
+
**Use predefined Accelerate configs:**
|
|
229
|
+
|
|
230
|
+
TRL provides predefined configs: `single_gpu`, `multi_gpu`, `fsdp1`, `fsdp2`, `zero1`, `zero2`, `zero3`
|
|
231
|
+
|
|
232
|
+
```bash
|
|
233
|
+
trl sft \
|
|
234
|
+
--config sft_config.yaml \
|
|
235
|
+
--accelerate_config zero2
|
|
236
|
+
```
|
|
237
|
+
|
|
238
|
+
**Custom Accelerate config:**
|
|
239
|
+
|
|
240
|
+
```bash
|
|
241
|
+
# Generate custom config
|
|
242
|
+
accelerate config
|
|
243
|
+
|
|
244
|
+
# Use custom config
|
|
245
|
+
trl sft --config sft_config.yaml --config_file ~/.cache/huggingface/accelerate/default_config.yaml
|
|
246
|
+
```
|
|
247
|
+
|
|
248
|
+
**Fully Sharded Data Parallel (FSDP):**
|
|
249
|
+
|
|
250
|
+
```bash
|
|
251
|
+
trl sft --config sft_config.yaml --accelerate_config fsdp2
|
|
252
|
+
```
|
|
253
|
+
|
|
254
|
+
**DeepSpeed ZeRO:**
|
|
255
|
+
|
|
256
|
+
```bash
|
|
257
|
+
trl sft --config sft_config.yaml --accelerate_config zero3
|
|
258
|
+
```
|
|
259
|
+
|
|
260
|
+
## Troubleshooting
|
|
261
|
+
|
|
262
|
+
### CUDA Out of Memory
|
|
263
|
+
|
|
264
|
+
- Reduce `--per_device_train_batch_size` and increase `--gradient_accumulation_steps`
|
|
265
|
+
- Enable `--use_peft` for LoRA training
|
|
266
|
+
- Use `--gradient_checkpointing` to save memory
|
|
267
|
+
- Try smaller model or longer sequence truncation
|
|
268
|
+
|
|
269
|
+
### Dataset Loading Issues
|
|
270
|
+
|
|
271
|
+
- Verify dataset exists: check Hugging Face Hub or local path
|
|
272
|
+
- Check dataset format matches expected columns
|
|
273
|
+
- Use `--dataset_config` for multi-config datasets
|
|
274
|
+
- Inspect dataset: `from datasets import load_dataset; ds = load_dataset(name)`
|
|
275
|
+
|
|
276
|
+
### Model Loading Issues
|
|
277
|
+
|
|
278
|
+
- Verify model exists on Hugging Face Hub
|
|
279
|
+
- Check if gated model requires authentication: `hf auth login`
|
|
280
|
+
- For local models, provide absolute path
|
|
281
|
+
- Ensure sufficient disk space and memory
|
|
282
|
+
|
|
283
|
+
### Slow Training
|
|
284
|
+
|
|
285
|
+
- Enable dataset `--packing` for short sequences
|
|
286
|
+
- Use larger `--per_device_train_batch_size` if memory allows
|
|
287
|
+
- Enable `--tf32` for faster computation on Ampere GPUs
|
|
288
|
+
- Use `--bf16` on supported hardware
|
|
289
|
+
- Consider multi-GPU training with `--num_processes`
|
|
290
|
+
|
|
291
|
+
### Generation Issues (GRPO/RLOO)
|
|
292
|
+
|
|
293
|
+
- Check prompt format in dataset
|
|
294
|
+
- Adjust `--temperature` and `--top_p` for generation
|
|
295
|
+
- Verify the reward function (for GRPO/RLOO)
|
|
296
|
+
|
|
297
|
+
## Additional Resources
|
|
298
|
+
|
|
299
|
+
- **Documentation**: https://huggingface.co/docs/trl
|
|
300
|
+
- **GitHub**: https://github.com/huggingface/trl
|
|
301
|
+
- **Examples**: https://github.com/huggingface/trl/tree/main/examples
|
|
302
|
+
|
|
303
|
+
## Best Practices
|
|
304
|
+
|
|
305
|
+
1. **Start with SFT**: Always fine-tune base models with SFT before preference alignment
|
|
306
|
+
2. **Use LoRA for efficiency**: Enable `--use_peft` for faster training and lower memory
|
|
307
|
+
3. **Monitor training**: Use `--report_to trackio` (or `--report_to wandb` or `--report_to tensorboard`) for tracking
|
|
308
|
+
4. **Save checkpoints**: TRL automatically saves checkpoints in `--output_dir`
|
|
309
|
+
5. **Test on small datasets first**: Verify pipeline works before full training
|
|
310
|
+
6. **Use configuration files**: Create YAML configs for reproducibility
|
|
311
|
+
7. **Leverage Accelerate**: Use multi-GPU training for faster iteration
|
|
312
|
+
|
|
313
|
+
When helping users with TRL:
|
|
314
|
+
- Always check which training method is appropriate for their use case
|
|
315
|
+
- Verify dataset format matches the expected schema
|
|
316
|
+
- Recommend starting with smaller models for testing
|
|
317
|
+
- Suggest LoRA for resource-constrained environments
|
|
318
|
+
- Point to specific documentation sections for advanced features
|