Downloads · 30 days
18
1% of all-time downloads
tog/TinyLlama-1.1B-alpaca-chat-v1.5
TinyLlama-1.1B-alpaca-chat-v1.5 is a text generation model from tog. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as apache-2.0.
This is the chat model finetuned on top of PY007/TinyLlama-1.1B-intermediate-step-715k-1.5T. The dataset used is tatsu-lab/stanfordalpaca.
Downloads · 30 days
18
1% of all-time downloads
All-time downloads
2.8K
Public
Repo size
8.8 GB
Likes
5
Public
Click a slice to open those files.
.bin4.4 GB · 100%
From the Hugging Face model README
This is the chat model finetuned on top of PY007/TinyLlama-1.1B-intermediate-step-715k-1.5T. The dataset used is tatsu-lab/stanford_alpaca.
Below is an instruction that describes a task. Write a response that appropriately completes the request.
### Instruction:
{instruction}
### Response:
You can use it with the transformers library:
from transformers import AutoTokenizer
import transformers
import torch
model = "tog/TinyLlama-1.1B-alpaca-chat-v1.5"
tokenizer = AutoTokenizer.from_pretrained(model)
pipeline = transformers.pipeline(
"text-generation",
model=model,
torch_dtype=torch.float16,
device_map="auto")
sequences = pipeline(
'###Instruction:\nWhat is a large language model? Be concise.\n\n### Response:\n',
do_sample=True,
top_k=10,
num_return_sequences=1,
eos_token_id=tokenizer.eos_token_id,
max_length=200)
for seq in sequences:
print(f"{seq['generated_text']}")
You should get something along those lines:
Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.
Result: ###Instruction:
What is a large language model? Be concise.
### Response:
A large language model is a type of natural language understanding model that can learn to accurately recognize and interpret text data by understanding the context of words. Languages used for text understanding are typically trained on a corpus of text data.