Update dam/model/language_model/builder.py

dam/model/language_model/builder.py

@@ -73,9 +73,12 @@ def build_llm_and_tokenizer(
     llm_cfg.model_max_length = model_max_length
     if model_max_length is not None:
         context_length_extension(llm_cfg)
-
+#  model_name_or_path, config=llm_cfg, torch_dtype=eval(config.model_dtype), *args, **kwargs
     llm = AutoModelForCausalLM.from_pretrained(
-        model_name_or_path, config=llm_cfg, torch_dtype=eval(config.model_dtype), *args, **kwargs
+       
+         llm_cfg["model_name_or_path"],
+    device_map="cpu",
+    torch_dtype=torch.float32  # 避免使用 fp16 权重导致报错
     )
     
     llm_path = model_name_or_path