Update llm.py

llm.py

@@ -4,17 +4,17 @@ login(token=os.getenv("HUGGINGFACE_HUB_TOKEN"))
 import asyncio
 from functools import partial
 import time
-from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
-
+from transformers import AutoProcessor, AutoModelForImageTextToText
 model_name = "Qwen/Qwen3.5-9B-Base"
 
-tokenizer = AutoTokenizer.from_pretrained(
+
+processor = AutoProcessor.from_pretrained(
     model_name,
     trust_remote_code=True
 )
 
-model = AutoModelForCausalLM.from_pretrained(
+model = AutoModelForImageTextToText.from_pretrained(
     model_name,
     device_map="auto",
     torch_dtype=torch.float16,