safely remove text_model loading in weights.py

weights.py

@@ -162,24 +162,6 @@ def load_weights_from_safetensors(weights_file: str, model: nn.Module) -> None:
     with safetensors_open(weights_file) as get_tensor:
         all_keys = get_tensor.keys()
 
-        # is_quantized = any(
-        #     ".qweight" in key or "_quantized" in key or "quant." in key
-        #     for key in all_keys
-        # )
-
-        if "text_model.transformer.h.0.ln.weight" in all_keys:
-            layernorm_dtype = get_tensor("text_model.transformer.h.0.ln.weight").dtype
-        else:
-            layernorm_dtype = torch.float16
-
-        # linear_dtype = torch.int8 if is_quantized else torch.float16
-
-        model.text = build_text_model(
-            TextConfig
-        )
-        if model.setup_caches_flag:
-            model._setup_caches()
-
         if (
             "vision.blocks.0.attn.proj.bias" in all_keys
             or "model.vision.blocks.0.attn.proj.bias" in all_keys
@@ -201,22 +183,6 @@ def load_weights_from_pt(weights_file: str, model: nn.Module) -> None:
     """Load weights from a PyTorch file into a MoondreamModel instance."""
     tensors = torch.load(weights_file, map_location="cpu", weights_only=True)
     all_keys = tensors.keys()
-    # is_quantized = any(
-    #     ".qweight" in key or "_quantized" in key or "quant." in key for key in all_keys
-    # )
-
-    if "text.blocks.0.ln.weight" in all_keys:
-        layernorm_dtype = tensors["text.blocks.0.ln.weight"].dtype
-    else:
-        layernorm_dtype = torch.float16
-
-    # linear_dtype = torch.int8 if is_quantized else torch.float16
-    model.text = build_text_model(
-        TextConfig
-    )
-    if model.setup_caches_flag:
-        model._setup_caches()
-
     if (
         "vision.blocks.0.attn.proj.bias" in all_keys
         or "model.vision.blocks.0.attn.proj.bias" in all_keys