Astrixnet
/

cilo

Text Generation

Model card Files Files and versions

cilo / modeling_cilo.py

Kai

Upload modeling_cilo.py with huggingface_hub

1872d4d verified about 17 hours ago

History Blame Contribute Delete

360 Bytes

	from transformers.models.mistral.modeling_mistral import MistralForCausalLM, MistralModel
	from transformers.models.mistral.configuration_mistral import MistralConfig

	class CiloConfig(MistralConfig):
	model_type = "cilo"

	class CiloModel(MistralModel):
	config_class = CiloConfig

	class CiloForCausalLM(MistralForCausalLM):
	config_class = CiloConfig