Update app.py

app.py

@@ -11,22 +11,24 @@ import os
 
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
-"""
-def install_apex_without_cuda():
+def install_apex():
     cmd = [
         'pip', 'install', '-v', '--disable-pip-version-check', '--no-cache-dir',
-        'git+https://github.com/NVIDIA/apex.git'
+        '--no-build-isolation', '--config-settings', '--build-option=--cpp_ext',
+        '--config-settings', '--build-option=--cuda_ext', 'git+https://github.com/NVIDIA/apex.git'
     ]
-    subprocess.run(cmd, check=True)
+    try:
+        subprocess.run(cmd, check=True)
+    except subprocess.CalledProcessError as e:
+        print(f"Apex installation failed: {e}")
+        raise
 
 try:
     import apex
 except ModuleNotFoundError:
-    print("Apex not found, installing without CUDA extensions...")
-    install_apex_without_cuda()
-    # After installation, try to import apex again
-    import apex
-"""
+    print("Apex not found, installing...")
+    install_apex()
+    import apex  # Try to import again after installing
     
 def download_t5_model(model_id, save_directory):
     # Modelin tokenizer'ını ve modeli indir