Update qwen2_5_1_5b_kl16_mar_h.py

qwen2_5_1_5b_kl16_mar_h.py

@@ -3,15 +3,15 @@ import torch
 from transformers import AutoModelForCausalLM
 from src.models.mar.mar import mar_huge, mar_max
 from src.models.mar.vae import AutoencoderKL
-from src.models.harmon_siglip import Harmon
+from src.models.skywork_unipic_siglip import SkyworkUnipic
 from xtuner.utils import PROMPT_TEMPLATE
 from transformers import AutoTokenizer
 from transformers import SiglipVisionModel
 
 
 
-llm_name_or_path = "/mnt/datasets_vlm/models/Qwen2.5-1.5B-Instruct"
-siglip2_path = "/mnt/data_vlm/wangxiaokun/Unified/harmon/siglip2-so400m-patch16-512"
+llm_name_or_path = "Qwen/Qwen2.5-1.5B-Instruct"
+siglip2_path = "google/siglip2-so400m-patch16-512"
 
 
 prompt_template = dict(
@@ -31,14 +31,14 @@ tokenizer = dict(
 )
 
 model = dict(
-    type=Harmon,
+    type=SkyworkUnipic,
     tokenizer=tokenizer,
     prompt_template=prompt_template,
     vae=dict(
         type=AutoencoderKL,
         embed_dim=16,
         ch_mult=(1, 1, 2, 2, 4),
-        ckpt_path="/mnt/data_vlm/wangxiaokun/Unified/harmon/kl16.ckpt",
+        ckpt_path="checkpoint/kl16.ckpt",
     ),
     vae_scale=0.2325,
     llm=dict(