isek-ai
/

LightNovel-Intro-RetNet-400M

Text Generation

Generated from Trainer

Model card Files Files and versions Community

LightNovel-Intro-RetNet-400M / modeling_retnet.py

p1atdev's picture

include retnet modeling

ab30eb6 about 1 year ago

231 Bytes

	from RetNet.retnet.modeling_retnet import RetNetForCausalLM, RetNetConfig


	LightNovelIntroRetNetModelConfig = RetNetConfig


	class LightNovelIntroRetNetModel(RetNetForCausalLM):
	config_class = LightNovelIntroRetNetModelConfig