formatron.integrations.RWKV.PIPELINE Class Reference

A wrapper for the pipeline of RWKV. More...

Inheritance diagram for formatron.integrations.RWKV.PIPELINE:

Public Member Functions
	__init__ (self, model, WORD_NAME, FormatterBuilder formatter_builder=None)
	generate (self, ctx, token_count=100, args=PIPELINE_ARGS(), callback=None, state=None)

Public Attributes
	formatter

Detailed Description

A wrapper for the pipeline of RWKV.

Definition at line 47 of file RWKV.py.

Constructor & Destructor Documentation

__init__()

formatron.integrations.RWKV.PIPELINE.__init__	(		self,
			model,
			WORD_NAME,
		FormatterBuilder	formatter_builder = None )

Definition at line 49 of file RWKV.py.

Member Function Documentation

generate()

formatron.integrations.RWKV.PIPELINE.generate	(		self,
			ctx,
			token_count = 100,
			args = PIPELINE_ARGS(),
			callback = None,
			state = None )

Definition at line 58 of file RWKV.py.

Member Data Documentation

formatter

formatron.integrations.RWKV.PIPELINE.formatter

Definition at line 54 of file RWKV.py.

---

The documentation for this class was generated from the following file:

• src/formatron/integrations/RWKV.py