Skip to content

Commit c1322a2

Browse files
vmoensvmoens
authored
[Feature] LLMMaskedCategorical (#3041)
1 parent 7691318 commit c1322a2

File tree

5 files changed

+1100
-14
lines changed

5 files changed

+1100
-14
lines changed

‎docs/source/reference/modules.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -440,6 +440,7 @@ Some distributions are typically used in RL scripts.
440440
TruncatedNormal
441441
TanhDelta
442442
OneHotCategorical
443+
LLMMaskedCategorical
443444
MaskedCategorical
444445
MaskedOneHotCategorical
445446
Ordinal

0 commit comments

Comments
 (0)