lehduong commited on
Commit
ee6e68b
1 Parent(s): 4251aa7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import spaces
2
  import subprocess
3
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
4
- subprocess.run("pip install --force-reinstall 'https://github.com/bitsandbytes-foundation/bitsandbytes/releases/download/continuous-release_multi-backend-refactor/bitsandbytes-0.44.1.dev0-py3-none-manylinux_2_24_x86_64.whl'", shell=True)
5
  import gradio as gr
6
  import torch
7
  import base64
@@ -713,7 +712,7 @@ with gr.Blocks(title="OneDiffusion Demo") as demo:
713
 
714
  if __name__ == "__main__":
715
  parser = argparse.ArgumentParser(description='Start the Gradio demo with specified captioner.')
716
- parser.add_argument('--captioner', type=str, choices=['molmo', 'llava', 'disable'], default='molmo', help='Captioner to use: molmo, llava, disable.')
717
  args = parser.parse_args()
718
 
719
  # Initialize models with the specified captioner
 
1
  import spaces
2
  import subprocess
3
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
4
  import gradio as gr
5
  import torch
6
  import base64
 
712
 
713
  if __name__ == "__main__":
714
  parser = argparse.ArgumentParser(description='Start the Gradio demo with specified captioner.')
715
+ parser.add_argument('--captioner', type=str, choices=['molmo', 'llava', 'disable'], default='disable', help='Captioner to use: molmo, llava, disable.')
716
  args = parser.parse_args()
717
 
718
  # Initialize models with the specified captioner