How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="Catniti/catrex-1.0")
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("Catniti/catrex-1.0")
model = AutoModelForCausalLM.from_pretrained("Catniti/catrex-1.0", device_map="auto")
messages = [
    {"role": "user", "content": "Who are you?"},
]
inputs = tokenizer.apply_chat_template(
	messages,
	add_generation_prompt=True,
	tokenize=True,
	return_dict=True,
	return_tensors="pt",
).to(model.device)

outputs = model.generate(**inputs, max_new_tokens=40)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))
Quick Links

Catrex 1.0

Bilingual (RU/EN) chat model on 65M parameters, trained from scratch.

Results

Metric Value
val loss 4.6802
perplexity 107.8
Parameters 64.82M
Context 512 tokens
Training steps 430

Run in LM Studio

  1. Download catrex-1.0-f16.gguf
  2. Put it in ~/.lmstudio/models/Catniti/catrex-1.0/
  3. Load it in LM Studio (chat template is already inside the file)

Run via llama.cpp

llama-cli -hf Catniti/catrex-1.0:f16 -p "Hi!"

Python (transformers)

from transformers import AutoModelForCausalLM, AutoTokenizer
tok = AutoTokenizer.from_pretrained("Catniti/catrex-1.0")
model = AutoModelForCausalLM.from_pretrained("Catniti/catrex-1.0")
msgs = [{"role": "user", "content": "Hi!"}]
ids = tok.apply_chat_template(msgs, add_generation_prompt=True, return_tensors="pt")
out = model.generate(ids, max_new_tokens=100, temperature=0.8, do_sample=True)
print(tok.decode(out[0][ids.shape[1]:], skip_special_tokens=True))

Prompt format

ChatML:

<|im_start|>user
Hi!<|im_end|>
<|im_start|>assistant
Downloads last month
968
Safetensors
Model size
64.8M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 1 Ask for provider support

Datasets used to train Catniti/catrex-1.0