Skip to content

Commit fe3e127

Browse files
Merge pull request #1572 from jiridanek/jd_document_trysty_torch
RHOAIENG-12554, RHOAIENG-12195: add TrustyAI `README.md` with TMaRCo detoxification details and required dependencies justification
2 parents a665a32 + f65d7d1 commit fe3e127

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

jupyter/trustyai/README.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
Supports the use of TrustyAI
2+
3+
Features:
4+
5+
* TMaRCo detoxification from the notebooks ([https://issues.redhat.com/browse/RHOAIENG-12554], [https://issues.redhat.com/browse/RHOAIENG-12195])
6+
* This uses an "expert"/"anti-expert" architecture, which requires running these models locally.
7+
This meant that these dependencies (Pytorch, transformers and datasets) had to be added to the dependencies.
8+
* The use-case benefits from GPU acceleration, so the CUDA-enabled version of Pytorch was chosen.

0 commit comments

Comments
 (0)