blob: 2759b4cbf9e72004f181ad2d937fc2ad851472af (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
pkgbase = python-trl
pkgdesc = Train transformer language models with reinforcement learning
pkgver = 0.13.0
pkgrel = 2
url = https://github.com/huggingface/trl
arch = any
groups = huggingface
license = Apache
makedepends = python-build
makedepends = python-installer
makedepends = python-setuptools
makedepends = python-wheel
depends = python-accelerate
depends = python-datasets
depends = python-rich
depends = python-transformers>=4.46.0
optdepends = python-bitsandbytes: Quantization support
optdepends = python-deepspeed: Low-level routines for distributed training
optdepends = python-diffusers: Diffusion models
optdepends = python-peft: Parameter efficient fine-tuning
source = python-trl-0.13.0.tar.gz::https://github.com/lvwerra/trl/archive/refs/tags/v0.13.0.tar.gz
sha256sums = 4a9e2dbd32d3fd993cf77493cfceec35b9d31dc51d90116b0a225ef4852210c7
pkgname = python-trl
|