Biomistral Instructv0.2 7B DARE

🚀 BioMistral-Instructv0.2-7B-DARE

BioMistral-Instructv0.2-7B-DARE 是一個融合模型，它使用 LazyMergekit 對以下模型進行了融合：

• BioMistral/BioMistral-7B

📦 安裝指南

!pip install -qU transformers accelerate

📚 詳細文檔

🧩 配置信息

models:
  - model: mistralai/Mistral-7B-Instruct-v0.2
    # 基礎模型無需參數
  - model: BioMistral/BioMistral-7B
    parameters:
      density: 0.5
      weight: 0.5
merge_method: dare_ties
base_model: mistralai/Mistral-7B-Instruct-v0.2
parameters:
  int8_mask: true
dtype: bfloat16

💻 使用示例

基礎用法

# 安裝必要的庫
!pip install -qU transformers accelerate

from transformers import AutoTokenizer
import transformers
import torch

# 定義模型名稱
model = "rangan2510/BioMistral-Instructv0.2-7B-DARE"
# 定義用戶消息
messages = [{"role": "user", "content": "What is a large language model?"}]

# 加載分詞器
tokenizer = AutoTokenizer.from_pretrained(model)
# 應用聊天模板生成提示
prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
# 創建文本生成管道
pipeline = transformers.pipeline(
    "text-generation",
    model=model,
    torch_dtype=torch.float16,
    device_map="auto",
)

# 生成文本
outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
# 打印生成的文本
print(outputs[0]["generated_text"])