akhaliq HF Staff commited on
Commit
52ed211
·
verified ·
1 Parent(s): 412e318

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -25,8 +25,13 @@ if not vibevoice_dir.exists():
25
  subprocess.run(['git', 'clone', 'https://github.com/microsoft/VibeVoice.git'], check=True)
26
  print("Installing VibeVoice...")
27
  subprocess.run(['pip', 'install', '-e', './VibeVoice'], check=True)
 
 
28
  print("Installing flash-attn...")
29
- subprocess.run(['pip', 'install', 'flash-attn', '--no-build-isolation'], check=True)
 
 
 
30
 
31
  # Add the VibeVoice directory to path
32
  import sys
 
25
  subprocess.run(['git', 'clone', 'https://github.com/microsoft/VibeVoice.git'], check=True)
26
  print("Installing VibeVoice...")
27
  subprocess.run(['pip', 'install', '-e', './VibeVoice'], check=True)
28
+ print("Installing wheel (required for flash-attn)...")
29
+ subprocess.run(['pip', 'install', 'wheel'], check=True)
30
  print("Installing flash-attn...")
31
+ try:
32
+ subprocess.run(['pip', 'install', 'flash-attn', '--no-build-isolation'], check=True)
33
+ except subprocess.CalledProcessError:
34
+ print("Warning: flash-attn installation failed. Continuing without it...")
35
 
36
  # Add the VibeVoice directory to path
37
  import sys