Spaces:
Runtime error
Runtime error
Factor Studios
commited on
Update vision_analyzer.py
Browse files- vision_analyzer.py +5 -0
vision_analyzer.py
CHANGED
|
@@ -23,6 +23,11 @@ import torch
|
|
| 23 |
from transformers import AutoProcessor, AutoModelForCausalLM
|
| 24 |
# from transformers import AutoProcessor, AutoModelForCausalLM
|
| 25 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
# Attempt to install flash-attn
|
| 27 |
try:
|
| 28 |
subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, check=True, shell=True)
|
|
|
|
| 23 |
from transformers import AutoProcessor, AutoModelForCausalLM
|
| 24 |
# from transformers import AutoProcessor, AutoModelForCausalLM
|
| 25 |
|
| 26 |
+
# Ensure unrar is installed (for Hugging Face Spaces)
|
| 27 |
+
import os
|
| 28 |
+
if not os.path.exists("/usr/bin/unrar"):
|
| 29 |
+
os.system("apt-get update && apt-get install -y unrar")
|
| 30 |
+
|
| 31 |
# Attempt to install flash-attn
|
| 32 |
try:
|
| 33 |
subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, check=True, shell=True)
|