formatron.integrations.RWKV.PIPELINE_ARGS Class Reference

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

Inheritance diagram for formatron.integrations.RWKV.PIPELINE_ARGS:

Public Member Functions
	__init__ (self, temperature=1.0, top_p=0.2, top_k=0, alpha_frequency=0.2, alpha_presence=0.2, alpha_decay=0.996, token_ban=[], token_stop=[], chunk_len=256, engine_gen_config=EngineGenerationConfig())

Public Attributes
	engine_gen_config

Detailed Description

A wrapper for the arguments of the pipeline of RWKV.

Definition at line 16 of file RWKV.py.

Constructor & Destructor Documentation

__init__()

formatron.integrations.RWKV.PIPELINE_ARGS.__init__	(		self,
			temperature = 1.0,
			top_p = 0.2,
			top_k = 0,
			alpha_frequency = 0.2,
			alpha_presence = 0.2,
			alpha_decay = 0.996,
			token_ban = [],
			token_stop = [],
			chunk_len = 256,
			engine_gen_config = EngineGenerationConfig() )

Definition at line 18 of file RWKV.py.

Member Data Documentation

engine_gen_config

formatron.integrations.RWKV.PIPELINE_ARGS.engine_gen_config

Definition at line 31 of file RWKV.py.

---

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

• src/formatron/integrations/RWKV.py