Update app.py

app.py

@@ -13,11 +13,9 @@ tokenizer = AutoTokenizer.from_pretrained(base_model_name)
 
 # 加载基础模型到 CPU
 base_model = AutoModelForCausalLM.from_pretrained(base_model_name, torch_dtype=torch.float32, device_map=None)
-base_model.to("cpu")  # 确保在 CPU 上运行
 
 # 加载 LoRA 适配器
 model = PeftModel.from_pretrained(base_model, adapter_model_name)
-model.to("cpu")  # 确保适配器也在 CPU 上运行
 
 # 运行测试
 def chat_with_model(prompt):