Spaces:

chunking-ai
/

smoldocling-preview

Paused

taprosoft commited on 6 days ago

Commit

8bd893c

verified ·

1 Parent(s): 9aac05b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,11 +4,7 @@ import subprocess
 os.system("apt-get update")
 os.system("apt-get install --no-install-recommends -y poppler-utils tesseract-ocr")
-subprocess.run(
-    "pip3 install flash-attn --no-build-isolation",
-    env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
-    shell=True,
-)
 import time

 os.system("apt-get update")
 os.system("apt-get install --no-install-recommends -y poppler-utils tesseract-ocr")
+os.system("FLASH_ATTENTION_SKIP_CUDA_BUILD=true pip3 install flash-attn --no-build-isolation")
 import time