Dedicated Endpoints

Run this model inference on single tenant GPU with unmatched speed and reliability at scale.

Learn more
Container

Run this model inference with full control and performance in your environment.

Learn more

Get help setting up a custom Dedicated Endpoints.

Talk with our engineer to get a quote for reserved GPU instances with discounts.

README

License: mit

uyu

This is a nanoGPT checkpoint converted to a Hugging Face GPT-2-compatible GPT2LMHeadModel.

The model weights load with:

python

from transformers import AutoTokenizer, GPT2LMHeadModel, pipeline
model = GPT2LMHeadModel.from_pretrained(".")
tokenizer = AutoTokenizer.from_pretrained(".", trust_remote_code=True)
pipe = pipeline(
"text-generation",
model="mente-ai/uyu-1-10M",
trust_remote_code=True,
)

Use eos_token_id=3 during generation to stop at <STORY_END>.

The tokenizer is a SentencePiece model stored as uyu.model. It is not a standard GPT-2 byte-level BPE tokenizer.

Original checkpoint: ckpt.pt

Model provider

mente-ai

Model tree

Base

this model

Modalities

Input

Text

Output

Text

Pricing

Dedicated Endpoints

View details

Supported Functionality

Model APIs

Dedicated Endpoints

Container

More information

Explore FriendliAI today