PixieZehirNano / README.md
Maani's picture
Update README.md
17fefd3 verified
|
raw
history blame
1.36 kB
metadata
language:
  - en
  - es
  - fr
  - fa
  - hi
  - th
library_name: transformers
license: apache-2.0
pipeline_tag: text-generation

Model Card for Model ID

image/png

Introducing Pixie Zehir Nano.

Excelling in writing.

A fine tune of LLAMA 3.2 1B on HQ DATA™ from Pixie Zehir.

Model Details

  • Developed by: [Maani x BLNKBLK]
  • Language(s) (NLP): [English, Spanish, French, Persian, Hindi, Thai]
  • License: [Apache 2.0]
  • Finetuned from model : [h2oai/h2o-danube-1.8b-chat]

Agreements

Model is created for research purposes, it can and will hallucinate, use with caution.

Usage

pip install transformers==4.45.0
import torch
from transformers import pipeline
pipe = pipeline(
    "text-generation",
    model="Maani/PixieZehirNano",
    torch_dtype=torch.bfloat16,
    device_map="auto",
)

messages = [
    {"role": "user", "content": "Write a haiku."},
]
prompt = pipe.tokenizer.apply_chat_template(
    messages,
    tokenize=False,
    add_generation_prompt=True,
)
res = pipe(
    prompt,
    max_new_tokens=256,
)
print(res[0]["generated_text"])
# <|prompt|>Write a haiku.</s><|answer|> In the windowless room, Digital dreams consume, Unseen sun sets on a white rabbit's ears: [...]