mrfakename's picture
Update README.md
dfb4b66 verified
metadata
license: mit
base_model:
  - ServiceNow-AI/Apriel-5B-Base
pipeline_tag: text-generation
library_name: transformers

Apriel 5B Base (Llamafied)

An approximate conversion of Apriel 5B Base to the Llama format.

  • No trust_remote_code required
  • Compatible w/ popular fine-tuning frameworks
  • Easier to run

Usage:

from transformers import pipeline
import torch

device = 'cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu'

pipe = pipeline('text-generation', model='mrfakename/Apriel-5B-Base-llamafied', device=device)

print(pipe('It was a dark and stormy night. The wind')[0]['generated_text'])

License: MIT (same as the original Apriel 5B model)