frankleeeee
commited on
Commit
•
86cedd4
1
Parent(s):
984abfc
update
Browse files
app.py
CHANGED
@@ -42,16 +42,15 @@ def install_dependencies(enable_optimization=False):
|
|
42 |
except (ImportError, ModuleNotFoundError):
|
43 |
return False
|
44 |
|
45 |
-
if enable_optimization
|
46 |
-
#
|
47 |
-
# thus, we need to install it no matter what
|
48 |
if not _is_package_available("flash_attn"):
|
49 |
subprocess.run(
|
50 |
f"{sys.executable} -m pip install flash-attn --no-build-isolation",
|
51 |
env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
|
52 |
shell=True,
|
53 |
)
|
54 |
-
|
55 |
# install apex for fused layernorm
|
56 |
if not _is_package_available("apex"):
|
57 |
subprocess.run(
|
@@ -184,6 +183,7 @@ from opensora.utils.inference_utils import (
|
|
184 |
prepare_multi_resolution_info,
|
185 |
refine_prompts_by_openai,
|
186 |
split_prompt,
|
|
|
187 |
)
|
188 |
from opensora.utils.misc import to_torch_dtype
|
189 |
|
@@ -513,13 +513,13 @@ def main():
|
|
513 |
with gr.Row():
|
514 |
with gr.Column():
|
515 |
prompt_text = gr.Textbox(label="Prompt", placeholder="Describe your video here", lines=4)
|
516 |
-
refine_prompt = gr.Checkbox(value=
|
517 |
-
random_prompt_btn = gr.Button("Random Prompt By GPT4o")
|
518 |
|
519 |
gr.Markdown("## Basic Settings")
|
520 |
resolution = gr.Radio(
|
521 |
choices=["144p", "240p", "360p", "480p", "720p"],
|
522 |
-
value="
|
523 |
label="Resolution",
|
524 |
)
|
525 |
aspect_ratio = gr.Radio(
|
|
|
42 |
except (ImportError, ModuleNotFoundError):
|
43 |
return False
|
44 |
|
45 |
+
if enable_optimization:
|
46 |
+
# install flash attention
|
|
|
47 |
if not _is_package_available("flash_attn"):
|
48 |
subprocess.run(
|
49 |
f"{sys.executable} -m pip install flash-attn --no-build-isolation",
|
50 |
env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
|
51 |
shell=True,
|
52 |
)
|
53 |
+
|
54 |
# install apex for fused layernorm
|
55 |
if not _is_package_available("apex"):
|
56 |
subprocess.run(
|
|
|
183 |
prepare_multi_resolution_info,
|
184 |
refine_prompts_by_openai,
|
185 |
split_prompt,
|
186 |
+
has_openai_key
|
187 |
)
|
188 |
from opensora.utils.misc import to_torch_dtype
|
189 |
|
|
|
513 |
with gr.Row():
|
514 |
with gr.Column():
|
515 |
prompt_text = gr.Textbox(label="Prompt", placeholder="Describe your video here", lines=4)
|
516 |
+
refine_prompt = gr.Checkbox(value=has_openai_key(), label="Refine prompt with GPT4o", interactive=has_openai_key())
|
517 |
+
random_prompt_btn = gr.Button("Random Prompt By GPT4o", interactive=has_openai_key())
|
518 |
|
519 |
gr.Markdown("## Basic Settings")
|
520 |
resolution = gr.Radio(
|
521 |
choices=["144p", "240p", "360p", "480p", "720p"],
|
522 |
+
value="240p",
|
523 |
label="Resolution",
|
524 |
)
|
525 |
aspect_ratio = gr.Radio(
|