Chattruth 7B

🚀 ChatTruth-7B

ChatTruth-7B 是在Qwen-VL的基礎上，採用精心設計的數據進行優化訓練得到的模型。相較於Qwen-VL，該模型在大分辨率上有顯著提升。同時，創新性地提出了Restore Module，大幅減少了大分辨率下的計算量。

🚀 快速開始

from transformers import AutoModelForCausalLM, AutoTokenizer
from transformers.generation import GenerationConfig
import torch
torch.manual_seed(1234)
model_path = 'ChatTruth-7B' # your downloaded model path.

tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)

# use cuda device
model = AutoModelForCausalLM.from_pretrained(model_path, device_map="cuda", trust_remote_code=True).eval()

model.generation_config = GenerationConfig.from_pretrained(model_path, trust_remote_code=True)
model.generation_config.top_p = 0.01

query = tokenizer.from_list_format([
    {'image': 'demo.jpeg'},
    {'text': '圖片中的文字是什麼'},
])
response, history = model.chat(tokenizer, query=query, history=None)
print(response)

# 昆明太厲害了

📦 安裝指南