Factor Studios commited on
Commit
017ed67
·
verified ·
1 Parent(s): eeb4e51

Update vision_analyzer.py

Browse files
Files changed (1) hide show
  1. 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)