Git 20

🚀 git_20

git_20 是一個經過微調的模型，它藉助多模態和多任務深度學習模型構建機器學習管道，旨在為學生的數學推理提供自動診斷反饋。該模型基於 Microsoft GIT 進行微調，研究人員可對其進行實驗和進一步微調，以助力構建能有效為學生數學推理提供自動診斷反饋的多模型。

🚀 快速開始

git_20 是使用 1 塊英偉達 A100 - 80G GPU 對 Microsoft GIT 進行微調得到的模型。我們從 300 萬份學生作業中提取了包含教師反饋的 10 萬份學生作業作為訓練數據，訓練數據分為學生作業的圖像部分和教師反饋的文本部分。git_20 由 18 層組成，擁有超過 1.7 億個參數，最多佔用 0.7GB 的磁盤空間。

💻 使用示例

基礎用法

from transformers import AutoModelForCausalLM
from transformers import AutoProcessor
from PIL import Image
model = AutoModelForCausalLM.from_pretrained("Fan21/git_20")
processor = AutoProcessor.from_pretrained("Fan21/git_20")

image_path ='Please enter the image address here'
image = Image.open(image_path)
width, height = image.size
display(image.resize((int(1 * width), int(1 * height))))
pixel_values = processor(images=image, return_tensors="pt").pixel_values
with torch.no_grad():
    outputs = model.generate(pixel_values=pixel_values, max_length=50)

answer = tokenizer.decode(outputs[0], skip_special_tokens=True)

📄 許可證

本項目採用 MIT 許可證。

📦 模型信息