Update src/models/mllm/seed_x.py

src/models/mllm/seed_x.py

@@ -4,7 +4,7 @@ import torch.nn as nn
 from torch.nn import functional as F
 from transformers import LogitsProcessorList
 from .generation import AutoImageTokenGenerationProcessor
-from .utils import load_zero3_checkpoint
+#from .utils import load_zero3_checkpoint
 
 
 BOI_TOKEN = '<img>'
@@ -234,5 +234,6 @@ class ContinuousLVLM(nn.Module):
 
         if pretrained_model_path is not None:
             ckpt = torch.load(pretrained_model_path, map_location='cpu')
-            load_zero3_checkpoint(model, ckpt)
+            missing, unexpected = model.load_state_dict(ckpt, strict=False)
+            #load_zero3_checkpoint(model, ckpt)
         return model