nemo-12b-story-v1/README.md

base_model, tags, license, language, library_name, pipeline_tag, model_type

base_model	tags	license	language	library_name	pipeline_tag	model_type
mistralai/Mistral-Nemo-Base-2407	mistral finetune text-generation	apache-2.0	en	transformers	text-generation	mistral

Nemo 12B Story v1

Fine-tuned Mistral-Nemo-12B for creative story generation.

Model Details

• Base Model: mistralai/Mistral-Nemo-Base-2407
• Architecture: MistralForCausalLM
• Parameters: 12B
• Context Length: 128k tokens
• Precision: BF16

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained("vclmax/nemo-12b-story-v1", torch_dtype="auto")
tokenizer = AutoTokenizer.from_pretrained("vclmax/nemo-12b-story-v1")