Skip to content

Commit f224f22

Browse files
laerdonlaerdon
andauthored
Add to transformers documentation (#296)
Co-authored-by: laerdon <lyk25@en-is-jacqueline.coecis.cornell.edu>
1 parent f38edf9 commit f224f22

2 files changed

Lines changed: 2 additions & 0 deletions

File tree

docs/source/transformerdecodermodel.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
Transformer Decoder-based Forecasting Model
22
===========================================
3+
Please note: this transformer requires installing convokit with the optional LLM packages via `pip install convokit[llm]`
34

45
A ConvoKit Forecaster-adherent implementation of conversational forecasting model based on Transformer Decoder Model (e.g. LlaMA, Gemma, GPT).
56
This class is first used in the paper "Conversations Gone Awry, But Then? Evaluating Conversational Forecasting Models" (Tran et al., 2025).

docs/source/transformerencodermodel.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
Transformer Encoder-based Forecasting Model
22
===========================================
3+
Please note: this transformer requires installing convokit with the optional LLM packages via `pip install convokit[llm]`.
34

45
A ConvoKit Forecaster-adherent implementation of conversational forecasting model based on Transformer Encoder Model (e.g. BERT, RoBERTa, SpanBERT, DeBERTa).
56
This class is first used in the paper "Conversations Gone Awry, But Then? Evaluating Conversational Forecasting Models"(Tran et al., 2025).

0 commit comments

Comments
 (0)