Spaces:
Build error
Build error
Update app.py
Browse files
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='
|
| 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
|