Skip to main content

TRL Jobs.

Project description

TRL Jobs

A convenient wrapper around hfjobs for running TRL (Transformer Reinforcement Learning) specific workflows on Hugging Face infrastructure.

Installation

pip install trl-jobs

Available Commands

SFT (Supervised Fine-Tuning)

Run SFT job with ease:

trl-jobs sft --flavor a100-large --model_name Qwen/Qwen3-0.6B --dataset_name trl-lib/Capybara

Required Arguments

  • --model_name: Model name (e.g., Qwen/Qwen3-0.6B)
  • --dataset_name: Dataset name (e.g., trl-lib/Capybara)

Optional Arguments

  • --flavor: Hardware flavor (default: t4-small)
  • -d, --detach: Run job in background and print job ID
  • --token: Hugging Face access token

Authentication

You can provide your Hugging Face token in several ways:

  1. Using huggingface-hub login: huggingface-cli login
  2. Setting the HF_TOKEN environment variable
  3. Using the --token argument

License

MIT License - see LICENSE file for details.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

trl_jobs-0.1.8.tar.gz (4.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

trl_jobs-0.1.8-py3-none-any.whl (7.7 kB view details)

Uploaded Python 3

File details

Details for the file trl_jobs-0.1.8.tar.gz.

File metadata

  • Download URL: trl_jobs-0.1.8.tar.gz
  • Upload date:
  • Size: 4.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for trl_jobs-0.1.8.tar.gz
Algorithm Hash digest
SHA256 c8c9212b3ddb49897a7df016c6f6ea6f7daf5926a0732cb0e59c562dcde8e705
MD5 e50c17f81b2fcefba9418825d2511a78
BLAKE2b-256 7199dec32ccae50a2d705d376e98456f7e64adaa3149b4d19355e9712140db44

See more details on using hashes here.

File details

Details for the file trl_jobs-0.1.8-py3-none-any.whl.

File metadata

  • Download URL: trl_jobs-0.1.8-py3-none-any.whl
  • Upload date:
  • Size: 7.7 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for trl_jobs-0.1.8-py3-none-any.whl
Algorithm Hash digest
SHA256 751b8482981b932a76f6e009e9e0851f1edbae8f0ec8b2e7069b94ef7fd58eca
MD5 459dca32e4e5b440d08bd8e5e3c54289
BLAKE2b-256 3f77e3b76f7146ad9b71d55ea04494adec2f37f98109314c9ad8a60938eba419

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page