metadata
license: apache-2.0
1. Model Details
Introducing xinchen9/Mistral-7B-CoT, an advanced language model comprising 8 billion parameters. It has been fine-trained based on mistralai/Mistral-7B-Instruct-v0.2.
The llama3-b8 model was fine-tuning on dataset CoT_ollection.
The training step is 10,000. The batch of each device is 16 and toal GPU is 5.
2. How to Use
Here give some examples of how to use our model.
Text Completion
import torch
from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
model_name = "xinchen9/Mistral-7B-CoT"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
model.generation_config = GenerationConfig.from_pretrained(model_name)
model.generation_config.pad_token_id = model.generation_config.eos_token_id
3 Disclaimer
The license on this model does not constitute legal advice. We are not responsible for the actions of third parties who use this model. Please cosult an attorney before using this model for commercial purposes.