Update main.py

main.py

@@ -13,7 +13,9 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 # 加载预训练模型
 model_name = "Qwen/Qwen2-0.5B"
 #model_name = "../models/qwen/Qwen2-0.5B"
-base_model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float32, device_map='auto')
+base_model = AutoModelForCausalLM.from_pretrained(model_name)
+
+base_model.to("cpu")
 
 # 加载适配器
 adapter_path1 = "test2023h5/wyw2xdw"