diff --git a/install_init.py b/install_init.py
index 21eba41..027a4af 100644
--- a/install_init.py
+++ b/install_init.py
@@ -69,7 +69,8 @@ def install_llama(system_info):
 
         if system_info['gpu']:
             cuda_version = system_info['cuda_version']
-            custom_command =  f"--force-reinstall --no-deps --index-url=https://abetlen.github.io/llama-cpp-python/whl/{cuda_version}"
+            #custom_command =  f"--force-reinstall --no-deps --index-url=https://abetlen.github.io/llama-cpp-python/whl/{cuda_version}"   #need fix 
+            custom_command =  f"--force-reinstall --no-deps --index-url=https://abetlen.github.io/llama-cpp-python/whl/cu124"
         else:
             custom_command = f"{base_url}{lcpp_version}/llama_cpp_python-{lcpp_version}-{system_info['platform_tag']}.whl"
         install_package("llama-cpp-python", custom_command=custom_command)