Skip to content

GitLab

Commit 66285be0 authored by zzg_666's avatar zzg_666
Browse files

zhaozg

parent 540877bf
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
Kimi_VL_think.py
View file @ 66285be0
...@@ -2,7 +2,7 @@ from PIL import Image ...@@ -2,7 +2,7 @@ from PIL import Image
import torch import torch
from transformers import AutoModelForCausalLM, AutoProcessor from transformers import AutoModelForCausalLM, AutoProcessor
model_path = "moonshotai/Kimi-VL-A3B-Thinking" model_path = "../moonshotai/Kimi-VL-A3B-Thinking"
model = AutoModelForCausalLM.from_pretrained( model = AutoModelForCausalLM.from_pretrained(
model_path, model_path,
torch_dtype=torch.bfloat16, torch_dtype=torch.bfloat16,
...@@ -20,7 +20,7 @@ model = AutoModelForCausalLM.from_pretrained( ...@@ -20,7 +20,7 @@ model = AutoModelForCausalLM.from_pretrained(
# ) # )
processor = AutoProcessor.from_pretrained(model_path, trust_remote_code=True) processor = AutoProcessor.from_pretrained(model_path, trust_remote_code=True)
image_paths = ["/xxx/Images/1.jpg", "/xxxx/Images/2.jpg"] image_paths = ["./Pic/arch.png", "./Pic/theory.png"]
images = [Image.open(path) for path in image_paths] images = [Image.open(path) for path in image_paths]
messages = [ messages = [
{ {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment