diff --git a/quantize/bnb_quantize.py b/quantize/bnb_quantize.py
index 205cd28..7aa7b46 100644
--- a/quantize/bnb_quantize.py
+++ b/quantize/bnb_quantize.py
@@ -18,11 +18,14 @@ import torch
 import GPUtil
 import os
 
+assert torch.cuda.is_available(),"CUDA is not available, but this code requires a GPU."
+
+device = 'cuda'  # Select GPU to use
 model_path = '/root/ld/ld_model_pretrained/MiniCPM-Llama3-V-2_5' # Model download path
-device = 'cuda' # Select GPU if available, otherwise CPU
 save_path = '/root/ld/ld_model_pretrain/MiniCPM-Llama3-V-2_5_int4' # Quantized model save path
 image_path = './assets/airplane.jpeg'
 
+
 # Create a configuration object to specify quantization parameters
 quantization_config = BitsAndBytesConfig(
     load_in_4bit=True,  # Whether to perform 4-bit quantization