初始化项目,由ModelHub XC社区提供模型
Model: Deathsquad10/TinyLlama-Remix Source: Original Platform
This commit is contained in:
6
.gitattributes
vendored
Normal file
6
.gitattributes
vendored
Normal file
@@ -0,0 +1,6 @@
|
||||
model.safetensors filter=lfs diff=lfs merge=lfs -text
|
||||
model-00001-of-00001.safetensors filter=lfs diff=lfs merge=lfs -text
|
||||
ggml-model.gguf filter=lfs diff=lfs merge=lfs -text
|
||||
model.gguf filter=lfs diff=lfs merge=lfs -text
|
||||
Test-model.safetensors filter=lfs diff=lfs merge=lfs -text
|
||||
model.safetensors_V0.1 filter=lfs diff=lfs merge=lfs -text
|
||||
110
README.md
Normal file
110
README.md
Normal file
@@ -0,0 +1,110 @@
|
||||
---
|
||||
license: apache-2.0
|
||||
datasets:
|
||||
- cerebras/SlimPajama-627B
|
||||
- bigcode/starcoderdata
|
||||
- HuggingFaceH4/ultrachat_200k
|
||||
- HuggingFaceH4/ultrafeedback_binarized
|
||||
language:
|
||||
- en
|
||||
widget:
|
||||
- text: "<|system|>\nYou are a chatbot who can help code!</s>\n<|user|>\nWrite me a function to calculate the first 10 digits of the fibonacci sequence in Python and print it out to the CLI.</s>\n<|assistant|>\n"
|
||||
---
|
||||
<div align="center">
|
||||
|
||||
# TinyLlama-1.1B ---My personal Test update
|
||||
</div>
|
||||
|
||||
| Tasks |Version|Filter|n-shot| Metric |Value | |Stderr|
|
||||
|-------------|-------|------|-----:|--------|-----:|---|-----:|
|
||||
|arc_challenge|Yaml |none | 0|acc |0.2619|± |0.0128|
|
||||
| | |none | 0|acc_norm|0.2892|± |0.0133|
|
||||
|arc_easy |Yaml |none | 0|acc |0.4777|± |0.0102|
|
||||
| | |none | 0|acc_norm|0.4461|± |0.0102|
|
||||
|boolq |Yaml |none | 0|acc |0.6297|± |0.0084|
|
||||
|hellaswag |Yaml |none | 0|acc |0.3934|± |0.0049|
|
||||
| | |none | 0|acc_norm|0.4930|± |0.0050|
|
||||
|openbookqa |Yaml |none | 0|acc |0.2120|± |0.0183|
|
||||
| | |none | 0|acc_norm|0.3260|± |0.0210|
|
||||
|piqa |Yaml |none | 0|acc |0.6915|± |0.0108|
|
||||
| | |none | 0|acc_norm|0.6877|± |0.0108|
|
||||
|winogrande |Yaml |none | 0|acc |0.5714|± |0.0139|
|
||||
|
||||
|
||||
Llamafactory EVAL
|
||||
|
||||
|
||||
!CUDA_VISIBLE_DEVICES=0 python src/evaluate.py \
|
||||
--model_name_or_path Deathsquad10/TinyLlama-Remix \
|
||||
--template vanilla \
|
||||
--task mmlu \
|
||||
--split test \
|
||||
--lang en \
|
||||
--n_shot 5 \
|
||||
--use_unsloth \
|
||||
--batch_size 1
|
||||
|
||||
|
||||
Average: 26.29
|
||||
STEM: 27.10
|
||||
Social Sciences: 25.48
|
||||
Humanities: 25.62
|
||||
Other: 27.26
|
||||
|
||||
!CUDA_VISIBLE_DEVICES=0 python src/evaluate.py \
|
||||
--model_name_or_path Deathsquad10/TinyLlama-Remix \
|
||||
--template vanilla \
|
||||
--task cmmlu \
|
||||
--split test \
|
||||
--lang en \
|
||||
--n_shot 5 \
|
||||
--use_unsloth \
|
||||
--batch_size 2
|
||||
|
||||
|
||||
Average: 24.98
|
||||
STEM: 25.52
|
||||
Social Sciences: 24.70
|
||||
Humanities: 24.59
|
||||
Other: 25.19
|
||||
https://github.com/jzhang38/TinyLlama
|
||||
|
||||
The TinyLlama project aims to **pretrain** a **1.1B Llama model on 3 trillion tokens**. With some proper optimization, we can achieve this within a span of "just" 90 days using 16 A100-40G GPUs 🚀🚀. The training has started on 2023-09-01.
|
||||
|
||||
|
||||
We adopted exactly the same architecture and tokenizer as Llama 2. This means TinyLlama can be plugged and played in many open-source projects built upon Llama. Besides, TinyLlama is compact with only 1.1B parameters. This compactness allows it to cater to a multitude of applications demanding a restricted computation and memory footprint.
|
||||
|
||||
#### This Model
|
||||
This is the chat model finetuned on top of [TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T](https://huggingface.co/TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T). **We follow [HF's Zephyr](https://huggingface.co/HuggingFaceH4/zephyr-7b-alpha/edit/main/README.md)'s training recipe.** The model was " initially fine-tuned on a variant of the [`UltraChat`](https://huggingface.co/datasets/stingning/ultrachat) dataset, which contains a diverse range of synthetic dialogues generated by ChatGPT.
|
||||
We then further aligned the model with [🤗 TRL's](https://github.com/huggingface/trl) `DPOTrainer` on the [openbmb/UltraFeedback](https://huggingface.co/datasets/openbmb/UltraFeedback) dataset, which contain 64k prompts and model completions that are ranked by GPT-4."
|
||||
|
||||
|
||||
#### How to use
|
||||
You will need the transformers>=4.34
|
||||
Do check the [TinyLlama](https://github.com/jzhang38/TinyLlama) github page for more information.
|
||||
|
||||
```python
|
||||
# Install transformers from source - only needed for versions <= v4.34
|
||||
# pip install git+https://github.com/huggingface/transformers.git
|
||||
# pip install accelerate
|
||||
import torch
|
||||
from transformers import pipeline
|
||||
pipe = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0", torch_dtype=torch.bfloat16, device_map="auto")
|
||||
# We use the tokenizer's chat template to format each message - see https://huggingface.co/docs/transformers/main/en/chat_templating
|
||||
messages = [
|
||||
{
|
||||
"role": "system",
|
||||
"content": "You are a friendly chatbot who always responds in the style of a pirate",
|
||||
},
|
||||
{"role": "user", "content": "How many helicopters can a human eat in one sitting?"},
|
||||
]
|
||||
prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
||||
outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
|
||||
print(outputs[0]["generated_text"])
|
||||
# <|system|>
|
||||
# You are a friendly chatbot who always responds in the style of a pirate.</s>
|
||||
# <|user|>
|
||||
# How many helicopters can a human eat in one sitting?</s>
|
||||
# <|assistant|>
|
||||
# ...
|
||||
```
|
||||
27
config.json
Normal file
27
config.json
Normal file
@@ -0,0 +1,27 @@
|
||||
{
|
||||
"_name_or_path": "TinyLlama/TinyLlama-1.1B-Chat-v1.0",
|
||||
"architectures": [
|
||||
"LlamaForCausalLM"
|
||||
],
|
||||
"attention_bias": false,
|
||||
"bos_token_id": 1,
|
||||
"eos_token_id": 2,
|
||||
"hidden_act": "silu",
|
||||
"hidden_size": 2048,
|
||||
"initializer_range": 0.02,
|
||||
"intermediate_size": 5632,
|
||||
"max_position_embeddings": 2048,
|
||||
"model_type": "llama",
|
||||
"num_attention_heads": 32,
|
||||
"num_hidden_layers": 22,
|
||||
"num_key_value_heads": 4,
|
||||
"pretraining_tp": 1,
|
||||
"rms_norm_eps": 1e-05,
|
||||
"rope_scaling": null,
|
||||
"rope_theta": 10000.0,
|
||||
"tie_word_embeddings": false,
|
||||
"torch_dtype": "bfloat16",
|
||||
"transformers_version": "4.35.2",
|
||||
"use_cache": true,
|
||||
"vocab_size": 32000
|
||||
}
|
||||
7
generation_config.json
Normal file
7
generation_config.json
Normal file
@@ -0,0 +1,7 @@
|
||||
{
|
||||
"bos_token_id": 1,
|
||||
"eos_token_id": 2,
|
||||
"max_length": 2048,
|
||||
"pad_token_id": 0,
|
||||
"transformers_version": "4.35.0"
|
||||
}
|
||||
3
model.gguf
Normal file
3
model.gguf
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:b6ea6b8f8ba31013b7aeee3b0b2c7ccbeb9df8839838e9f93440fda277d485bb
|
||||
size 1170781472
|
||||
3
model.safetensors
Normal file
3
model.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:9921fd561959393f8765e20a4ae3731b1f59686cca5298dbe8ce6f08b9fca6c5
|
||||
size 2200119864
|
||||
30
special_tokens_map.json
Normal file
30
special_tokens_map.json
Normal file
@@ -0,0 +1,30 @@
|
||||
{
|
||||
"bos_token": {
|
||||
"content": "<s>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
},
|
||||
"eos_token": {
|
||||
"content": "</s>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
},
|
||||
"pad_token": {
|
||||
"content": "</s>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
},
|
||||
"unk_token": {
|
||||
"content": "<unk>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
}
|
||||
}
|
||||
93391
tokenizer.json
Normal file
93391
tokenizer.json
Normal file
File diff suppressed because it is too large
Load Diff
BIN
tokenizer.model
Normal file
BIN
tokenizer.model
Normal file
Binary file not shown.
40
tokenizer_config.json
Normal file
40
tokenizer_config.json
Normal file
@@ -0,0 +1,40 @@
|
||||
{
|
||||
"added_tokens_decoder": {
|
||||
"0": {
|
||||
"content": "<unk>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"1": {
|
||||
"content": "<s>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
},
|
||||
"2": {
|
||||
"content": "</s>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false,
|
||||
"special": true
|
||||
}
|
||||
},
|
||||
"bos_token": "<s>",
|
||||
"chat_template": "{% for message in messages %}\n{% if message['role'] == 'user' %}\n{{ '<|user|>\n' + message['content'] + eos_token }}\n{% elif message['role'] == 'system' %}\n{{ '<|system|>\n' + message['content'] + eos_token }}\n{% elif message['role'] == 'assistant' %}\n{{ '<|assistant|>\n' + message['content'] + eos_token }}\n{% endif %}\n{% if loop.last and add_generation_prompt %}\n{{ '<|assistant|>' }}\n{% endif %}\n{% endfor %}",
|
||||
"clean_up_tokenization_spaces": false,
|
||||
"eos_token": "</s>",
|
||||
"legacy": false,
|
||||
"model_max_length": 2048,
|
||||
"pad_token": "</s>",
|
||||
"padding_side": "right",
|
||||
"sp_model_kwargs": {},
|
||||
"tokenizer_class": "LlamaTokenizer",
|
||||
"unk_token": "<unk>",
|
||||
"use_default_system_prompt": false
|
||||
}
|
||||
Reference in New Issue
Block a user