[Model] SiglipVisionModel ported from transformers (#6942)

Co-authored-by: Roger Wang <ywang@roblox.com>
This commit is contained in:
Jungho Christopher Cho
2024-08-05 15:22:12 +09:00
committed by GitHub
parent cc08fc7225
commit c0d8f1636c
3 changed files with 650 additions and 53 deletions

View File

@@ -65,7 +65,8 @@ def run_phi3v(question):
# PaliGemma
def run_paligemma(question):
prompt = question
# PaliGemma has special prompt format for VQA
prompt = "caption en"
llm = LLM(model="google/paligemma-3b-mix-224")
return llm, prompt